diff options
| -rw-r--r-- | environment.yaml | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/environment.yaml b/environment.yaml index 85033ce..cf2b732 100644 --- a/environment.yaml +++ b/environment.yaml | |||
| @@ -16,7 +16,7 @@ dependencies: | |||
| 16 | - python=3.10.8 | 16 | - python=3.10.8 |
| 17 | - pytorch=2.0.0=*cuda11.8* | 17 | - pytorch=2.0.0=*cuda11.8* |
| 18 | - torchvision=0.15.0 | 18 | - torchvision=0.15.0 |
| 19 | - xformers=0.0.19 | 19 | - xformers=0.0.20.dev528 |
| 20 | - pip: | 20 | - pip: |
| 21 | - -e . | 21 | - -e . |
| 22 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate | 22 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate |
| @@ -30,5 +30,5 @@ dependencies: | |||
| 30 | - safetensors==0.3.1 | 30 | - safetensors==0.3.1 |
| 31 | - setuptools==65.6.3 | 31 | - setuptools==65.6.3 |
| 32 | - test-tube>=0.7.5 | 32 | - test-tube>=0.7.5 |
| 33 | - timm==0.8.17.dev0 | 33 | - timm==0.9.2 |
| 34 | - transformers==4.29.0 | 34 | - transformers==4.29.1 |