diffusers
a59f9990 - Tests: upgrade PyTorch cuda to 11.7 to fix examples tests. (#1048)

Commit
2 years ago
Tests: upgrade PyTorch cuda to 11.7 to fix examples tests. (#1048) Tests: upgrade PyTorch cuda to 11.7. Otherwise the cuda versions of torch and torchvision mismatch, and examples tests fail. We were requesting cuda 11.6 for PyTorch, and the default torchvision (via setup.py). Another option would be to include torchvision in the same pip install line as torch.
Author
Parents
Loading