Rishi Desai commited on
Commit
5fb47d3
·
1 Parent(s): 2fa4b3f

listing model files

Browse files
Files changed (1) hide show
  1. 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
- # {"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,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()