使用CPU加载报错
#16
by
lanpangzi
- opened
File "C:\Users\lanpangzi\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\cuda_init_.py", line 284, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
modeling_minicpmo.py 第187行,删除.to("cuda")
vocos = Vocos(feature_extractor, backbone, head).eval().to(torch.float32)