Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -41,4 +41,5 @@ accelerate
|
|
41 |
git+https://github.com/huggingface/diffusers
|
42 |
einops
|
43 |
numpy==1.26.3
|
44 |
-
fvcore
|
|
|
|
41 |
git+https://github.com/huggingface/diffusers
|
42 |
einops
|
43 |
numpy==1.26.3
|
44 |
+
fvcore
|
45 |
+
pytorch-lightning==2.2.0
|