MattyAtHome/FluxD_Workflows
0
1import os2from huggingface_hub import hf_hub_download3import subprocess4 5# --- Ensure target directories exist (inside ComfyUI/models) ---6directories = [7 "ComfyUI/models/style_models",8 "ComfyUI/models/diffusion_models",9 "ComfyUI/models/clip_vision",10 "ComfyUI/models/depthanything",11 "ComfyUI/models/vae/FLUX1",12 "ComfyUI/models/clip",13 "ComfyUI/models/text_encoders",14 "ComfyUI/models/upscale_models",15 "ComfyUI/models/loras",16 "ComfyUI/user/default/workflows"17]18 19for d in directories:20 os.makedirs(d, exist_ok=True)21 22# --- Download workflow files ---23 24# Download FLUX_simpleGenerator into ComfyUI/user/default/workflows25hf_hub_download(26 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 27 filename="FLUX_simpleGenerator.json",28 local_dir="ComfyUI/user/default/workflows",29 #force_download=True30)31 32# Download FLUX_UpScale_dev into ComfyUI/user/default/workflows33hf_hub_download(34 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 35 filename="FLUX_UpScale_dev.json",36 local_dir="ComfyUI/user/default/workflows",37 #force_download=True38)39 40# Download FLUX_Fill_In-Outpainting into ComfyUI/user/default/workflows41hf_hub_download(42 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 43 filename="FLUX_Fill_In-Outpainting.json",44 local_dir="ComfyUI/user/default/workflows",45 #force_download=True46)47 48# Download FLUX_IMG2IMG into ComfyUI/user/default/workflows49hf_hub_download(50 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 51 filename="FLUX_IMG2IMG.json",52 local_dir="ComfyUI/user/default/workflows",53 #force_download=True54)55 56# Download FLUX_Ace_FaceSwap into ComfyUI/user/default/workflows57hf_hub_download(58 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 59 filename="FLUX_Ace_FaceSwap.json",60 local_dir="ComfyUI/user/default/workflows",61 #force_download=True62)63 64# --- end of workflows65 66 67# --- Download model files ---68 69# Download flux1-dev-fp8.safetensors into ComfyUI/models/diffusion_models70hf_hub_download(71 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 72 filename="flux1-dev-fp8.safetensors",73 local_dir="ComfyUI/models/diffusion_models",74 #force_download=True75)76 77# Download fluxFillFP8_v10.safetensors into ComfyUI/models/diffusion_models78hf_hub_download(79 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 80 filename="fluxFillFP8_v10.safetensors",81 local_dir="ComfyUI/models/diffusion_models",82 #force_download=True83)84 85# Download 4x-UltraSharp into ComfyUI/models/upscale_models86hf_hub_download(87 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 88 filename="4x-UltraSharp.pth",89 local_dir="ComfyUI/models/upscale_models",90 #force_download=True91)92 93# Download FLUX.1-Turbo-Alpha.safetensors into ComfyUI/models/loras94hf_hub_download(95 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 96 filename="FLUX.1-Turbo-Alpha.safetensors",97 local_dir="ComfyUI/models/loras",98 #force_download=True99)100 101# Download comfyui_portrait_lora64.safetensors into ComfyUI/models/loras102hf_hub_download(103 repo_id="MattyAtHome/DiffusionModelsAndDependencies", 104 filename="comfyui_portrait_lora64.safetensors",105 local_dir="ComfyUI/models/loras",106 #force_download=True107)108 109# Download clip_l.safetensors into ComfyUI/models/text_encoders110hf_hub_download(111 repo_id="MattyAtHome/DiffusionModelsAndDependencies",112 filename="clip_l.safetensors",113 local_dir="ComfyUI/models/text_encoders",114 #force_download=True115)116 117# Download t5xxl_fp16.safetensors into ComfyUI/models/text_encoders118hf_hub_download(119 repo_id="comfyanonymous/flux_text_encoders",120 filename="t5xxl_fp16.safetensors",121 local_dir="ComfyUI/models/text_encoders",122 #force_download=True123)124 125# Download t5xxl_fp8_e4m3fn.safetensors into ComfyUI/models/text_encoders126hf_hub_download(127 repo_id="comfyanonymous/flux_text_encoders",128 filename="t5xxl_fp8_e4m3fn.safetensors",129 local_dir="ComfyUI/models/text_encoders",130 #force_download=True131)132 133 134# Download ae.safetensors into ComfyUI/models/text_encoders135hf_hub_download(136 repo_id="MattyAtHome/DiffusionModelsAndDependencies",137 filename="ae.safetensors",138 local_dir="ComfyUI/models/vae",139 #force_download=True140)141 142# --- End of model file downloads143 144# --- Now that models are downloaded, start ComfyUI ---145print("Models downloaded, starting ComfyUI...")146subprocess.run(["python3", "ComfyUI/main.py", "--listen", "0.0.0.0", "--port", "7860"])