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
|