Open Kraiwin-Namsri opened 1 year ago
I got this error: " If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU. "
https://pytorch.org/docs/stable/generated/torch.load.html
when trying with this command: python demo.py --id May --driving_audio ./data/Input/00083.wav --device cp
python demo.py --id May --driving_audio ./data/Input/00083.wav --device cp
Explicitly adding this works: map_location=device
I am not able to try this on CUDA. Please check if this works on CUDA.
I got this error: " If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU. "
https://pytorch.org/docs/stable/generated/torch.load.html
when trying with this command:
python demo.py --id May --driving_audio ./data/Input/00083.wav --device cpExplicitly adding this works: map_location=device
I am not able to try this on CUDA. Please check if this works on CUDA.