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

tryign to remove flux as a test

Browse files
Files changed (1) hide show
  1. install.py +2 -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"},
@@ -214,6 +214,7 @@ def install(is_hf_space=False):
214
  install_comfyui()
215
  install_custom_nodes()
216
  if is_hf_space:
 
217
  install_hfdemo_dependencies()
218
  download_huggingface_models()
219
  download_and_extract_antelopev2()
 
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"},
 
214
  install_comfyui()
215
  install_custom_nodes()
216
  if is_hf_space:
217
+ print("🔄 Installing HF spaces dependencies...")
218
  install_hfdemo_dependencies()
219
  download_huggingface_models()
220
  download_and_extract_antelopev2()