Spaces:
Paused
Paused
Rishi Desai
commited on
Commit
·
5fb47d3
1
Parent(s):
2fa4b3f
listing model files
Browse files- install.py +4 -1
install.py
CHANGED
|
@@ -62,7 +62,7 @@ def download_huggingface_models():
|
|
| 62 |
"""Download required models from Hugging Face and symlink to ComfyUI models directory."""
|
| 63 |
from huggingface_hub import hf_hub_download
|
| 64 |
hf_models = [
|
| 65 |
-
|
| 66 |
{"repo_id": "black-forest-labs/FLUX.1-dev", "filename": "ae.safetensors", "folder": "vae"},
|
| 67 |
{"repo_id": "Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro", "filename": "diffusion_pytorch_model.safetensors",
|
| 68 |
"folder": "controlnet"},
|
|
@@ -219,6 +219,9 @@ def install(is_hf_space=False):
|
|
| 219 |
download_huggingface_models()
|
| 220 |
download_and_extract_antelopev2()
|
| 221 |
print("🎉 Setup Complete!")
|
|
|
|
|
|
|
|
|
|
| 222 |
|
| 223 |
if __name__ == "__main__":
|
| 224 |
install()
|
|
|
|
| 62 |
"""Download required models from Hugging Face and symlink to ComfyUI models directory."""
|
| 63 |
from huggingface_hub import hf_hub_download
|
| 64 |
hf_models = [
|
| 65 |
+
{"repo_id": "black-forest-labs/FLUX.1-dev", "filename": "flux1-dev.safetensors", "folder": "unet"},
|
| 66 |
{"repo_id": "black-forest-labs/FLUX.1-dev", "filename": "ae.safetensors", "folder": "vae"},
|
| 67 |
{"repo_id": "Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro", "filename": "diffusion_pytorch_model.safetensors",
|
| 68 |
"folder": "controlnet"},
|
|
|
|
| 219 |
download_huggingface_models()
|
| 220 |
download_and_extract_antelopev2()
|
| 221 |
print("🎉 Setup Complete!")
|
| 222 |
+
|
| 223 |
+
print("\n📂 Listing installed models:")
|
| 224 |
+
run_cmd("ls -Rh ./ComfyUI/models/")
|
| 225 |
|
| 226 |
if __name__ == "__main__":
|
| 227 |
install()
|