Upload environment.yml
Browse files- environment.yml +13 -0
environment.yml
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: DiT
|
| 2 |
+
channels:
|
| 3 |
+
- pytorch
|
| 4 |
+
- nvidia
|
| 5 |
+
dependencies:
|
| 6 |
+
- python >= 3.8
|
| 7 |
+
- pytorch >= 1.13
|
| 8 |
+
- torchvision
|
| 9 |
+
- pytorch-cuda=11.7
|
| 10 |
+
- pip:
|
| 11 |
+
- timm
|
| 12 |
+
- diffusers
|
| 13 |
+
- accelerate
|