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

Comment changes are shownComment changes are hidden
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
  • .github/workflows
    • File
      push_tests.yml