Dockerfile torch fix (#987)

* add torch to requirements.txt at build time to force version to stick

* fix xformers check

* better handling of xformers based on installed torch version

* fix for ci w/o torch
This commit is contained in:
Wing Lian
2023-12-21 09:38:20 -05:00
committed by GitHub
parent d25c34caa6
commit 161bcb6517
4 changed files with 12 additions and 10 deletions

View File

@@ -28,7 +28,7 @@ jobs:
- cuda: "118"
cuda_version: 11.8.0
python_version: "3.10"
pytorch: 2.1.0
pytorch: 2.1.1
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX"
steps:
- name: Checkout