Update requirements.txt
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
@@ -3,8 +3,8 @@ numpy
|
|
3 |
opencv-python
|
4 |
Pillow
|
5 |
torch
|
6 |
-
# torchvision # Marigold typically doesn't directly use torchvision, so it can be omitted
|
7 |
diffusers
|
8 |
huggingface_hub
|
9 |
-
transformers
|
10 |
-
|
|
|
|
3 |
opencv-python
|
4 |
Pillow
|
5 |
torch
|
|
|
6 |
diffusers
|
7 |
huggingface_hub
|
8 |
+
transformers
|
9 |
+
accelerate # <--- NEW: Added for memory efficiency and model loading
|
10 |
+
xformers # <--- Recommended for GPU performance with Marigold
|