Update requirements.txt
Browse files- requirements.txt +10 -5
requirements.txt
CHANGED
|
@@ -29,14 +29,19 @@ torchvision==0.20.1
|
|
| 29 |
torchaudio==2.5.1
|
| 30 |
transformers>=4.45.0,<5.0
|
| 31 |
huggingface_hub>=0.25.0,<1.0
|
| 32 |
-
|
|
|
|
| 33 |
safetensors==0.4.3
|
| 34 |
tokenizers>=0.19.0,<1.0
|
| 35 |
datasets>=2.14.0,<3.0
|
| 36 |
einops==0.8.0
|
| 37 |
timm>=0.9.16,<1.0
|
|
|
|
| 38 |
xformers>=0.0.28,<0.1.0
|
| 39 |
|
|
|
|
|
|
|
|
|
|
| 40 |
# ============================================================================
|
| 41 |
# COMPUTER VISION & VIDEO PROCESSING
|
| 42 |
# ============================================================================
|
|
@@ -98,7 +103,7 @@ marshmallow>=3.20.0,<4.0
|
|
| 98 |
cerberus>=1.3.5,<2.0
|
| 99 |
|
| 100 |
# ============================================================================
|
| 101 |
-
# MATANYONE (install from GitHub
|
| 102 |
# ============================================================================
|
| 103 |
git+https://github.com/pq-yang/MatAnyone.git
|
| 104 |
|
|
@@ -139,7 +144,7 @@ python-crontab>=3.0.0,<4.0
|
|
| 139 |
# NOTES / FIXES
|
| 140 |
# ============================================================================
|
| 141 |
# 1. Keep --extra-index-url at the top.
|
| 142 |
-
# 2. Install MatAnyone
|
| 143 |
# 3. Imageio pinned to 2.25.0 to avoid moviepy/scikit-image breakage.
|
| 144 |
-
# 4. If CUDA issues:
|
| 145 |
-
# 5.
|
|
|
|
| 29 |
torchaudio==2.5.1
|
| 30 |
transformers>=4.45.0,<5.0
|
| 31 |
huggingface_hub>=0.25.0,<1.0
|
| 32 |
+
# Bump Accelerate so Diffusers offloading & memory features are available
|
| 33 |
+
accelerate>=0.30.0,<1.0
|
| 34 |
safetensors==0.4.3
|
| 35 |
tokenizers>=0.19.0,<1.0
|
| 36 |
datasets>=2.14.0,<3.0
|
| 37 |
einops==0.8.0
|
| 38 |
timm>=0.9.16,<1.0
|
| 39 |
+
# xFormers optional; if no wheel for your env, Diffusers will still run without it
|
| 40 |
xformers>=0.0.28,<0.1.0
|
| 41 |
|
| 42 |
+
# >>> NEW: Diffusers for AI background generation <<<
|
| 43 |
+
diffusers>=0.30.0,<0.31
|
| 44 |
+
|
| 45 |
# ============================================================================
|
| 46 |
# COMPUTER VISION & VIDEO PROCESSING
|
| 47 |
# ============================================================================
|
|
|
|
| 103 |
cerberus>=1.3.5,<2.0
|
| 104 |
|
| 105 |
# ============================================================================
|
| 106 |
+
# MATANYONE (install from GitHub) -- NOTE: PyPI package not used
|
| 107 |
# ============================================================================
|
| 108 |
git+https://github.com/pq-yang/MatAnyone.git
|
| 109 |
|
|
|
|
| 144 |
# NOTES / FIXES
|
| 145 |
# ============================================================================
|
| 146 |
# 1. Keep --extra-index-url at the top.
|
| 147 |
+
# 2. Install MatAnyone from GitHub (PyPI version not used).
|
| 148 |
# 3. Imageio pinned to 2.25.0 to avoid moviepy/scikit-image breakage.
|
| 149 |
+
# 4. If CUDA issues: clear HF cache and restart Space.
|
| 150 |
+
# 5. Diffusers added for AI background creation; Accelerate bumped to enable memory offload features.
|