Spaces:
Running
on
Zero
Running
on
Zero
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()
|