使用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)

Sign up or log in to comment