diff options
Diffstat (limited to 'environment.yaml')
-rw-r--r-- | environment.yaml | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/environment.yaml b/environment.yaml index b161244..dfbafaf 100644 --- a/environment.yaml +++ b/environment.yaml | |||
@@ -4,7 +4,7 @@ channels: | |||
4 | - nvidia | 4 | - nvidia |
5 | - xformers/label/dev | 5 | - xformers/label/dev |
6 | - defaults | 6 | - defaults |
7 | - conda-forge | 7 | - conda-forge |
8 | dependencies: | 8 | dependencies: |
9 | - gcc=11.3.0 | 9 | - gcc=11.3.0 |
10 | - gxx=11.3.0 | 10 | - gxx=11.3.0 |
@@ -30,4 +30,3 @@ dependencies: | |||
30 | - test-tube>=0.7.5 | 30 | - test-tube>=0.7.5 |
31 | - timm==0.8.17.dev0 | 31 | - timm==0.8.17.dev0 |
32 | - transformers==4.28.1 | 32 | - transformers==4.28.1 |
33 | - triton==2.0.0.post1 | ||