diff options
Diffstat (limited to 'environment.yaml')
| -rw-r--r-- | environment.yaml | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/environment.yaml b/environment.yaml index 42b568f..9c12a0b 100644 --- a/environment.yaml +++ b/environment.yaml | |||
| @@ -11,6 +11,7 @@ dependencies: | |||
| 11 | - python=3.10.8 | 11 | - python=3.10.8 |
| 12 | - pytorch=2.0.0=*cuda11.8* | 12 | - pytorch=2.0.0=*cuda11.8* |
| 13 | - torchvision=0.15.0 | 13 | - torchvision=0.15.0 |
| 14 | - xformers=0.0.17.dev481 | ||
| 14 | - pip: | 15 | - pip: |
| 15 | - -e . | 16 | - -e . |
| 16 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 17 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
| @@ -24,4 +25,3 @@ dependencies: | |||
| 24 | - test-tube>=0.7.5 | 25 | - test-tube>=0.7.5 |
| 25 | - transformers==4.27.1 | 26 | - transformers==4.27.1 |
| 26 | - triton==2.0.0 | 27 | - triton==2.0.0 |
| 27 | - xformers==0.0.17.dev480 | ||