Spaces:
Sleeping
Sleeping
Commit
•
2c8c0a9
1
Parent(s):
a128a9a
Create environment.yml
Browse files- environment.yml +19 -0
environment.yml
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
channels:
|
| 2 |
+
- pytorch
|
| 3 |
+
- nvidia
|
| 4 |
+
dependencies:
|
| 5 |
+
- python=3.11.3
|
| 6 |
+
- pytorch=2.0.1
|
| 7 |
+
- pytorch-cuda=11.7
|
| 8 |
+
- torchvision=0.15.2
|
| 9 |
+
- pip:
|
| 10 |
+
- accelerate==0.19.0
|
| 11 |
+
- diffusers[torch]==0.16.0
|
| 12 |
+
- einops==0.6.1
|
| 13 |
+
- ffmpeg==1.4
|
| 14 |
+
- pandas==2.0.1
|
| 15 |
+
- tqdm==4.65.0
|
| 16 |
+
- transformers==4.28.1
|
| 17 |
+
- xformers==0.0.20
|
| 18 |
+
- rotary_embedding_torch
|
| 19 |
+
- gradio==4.3.0
|