juanmackie commited on
Commit
9bc73e7
·
verified ·
1 Parent(s): 4058b98

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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 # <--- CRUCIAL: Added or confirmed for Marigold's dependency
10
- xformers # <--- ADDED: For performance optimization with Marigold
 
 
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