diff --git a/.gitignore b/.gitignore index 5f0f0d9d36..204beca8c7 100644 --- a/.gitignore +++ b/.gitignore @@ -9,6 +9,7 @@ .server .vscode __pycache__ +scripts/upload.py scripts/docker/krita-ai-diffusion scripts/docker/ComfyUI scripts/downloads diff --git a/ai_diffusion/cloud_client.py b/ai_diffusion/cloud_client.py index f64cf7ae16..17e9c88216 100644 --- a/ai_diffusion/cloud_client.py +++ b/ai_diffusion/cloud_client.py @@ -410,7 +410,9 @@ def _checkpoint_info(id: str, arch: Arch): resource_id(ResourceKind.ip_adapter, Arch.sdxl, ControlMode.reference): "ip-adapter_sdxl_vit-h.safetensors", resource_id(ResourceKind.ip_adapter, Arch.sd15, ControlMode.face): None, resource_id(ResourceKind.ip_adapter, Arch.sdxl, ControlMode.face): None, + resource_id(ResourceKind.ip_adapter, Arch.flux, ControlMode.reference): "flux1-redux-dev.safetensors", resource_id(ResourceKind.clip_vision, Arch.all, "ip_adapter"): "clip-vision_vit-h.safetensors", + resource_id(ResourceKind.clip_vision, Arch.flux, "redux"): "sigclip_vision_patch14_384.safetensors", resource_id(ResourceKind.lora, Arch.sd15, "hyper"): "Hyper-SD15-8steps-CFG-lora.safetensors", resource_id(ResourceKind.lora, Arch.sdxl, "hyper"): "Hyper-SDXL-8steps-CFG-lora.safetensors", resource_id(ResourceKind.lora, Arch.sd15, ControlMode.face): None, diff --git a/ai_diffusion/resources.py b/ai_diffusion/resources.py index ff2fa5bbad..222c2a4c42 100644 --- a/ai_diffusion/resources.py +++ b/ai_diffusion/resources.py @@ -662,8 +662,8 @@ def __hash__(self): ResourceId(ResourceKind.controlnet, Arch.flux, ControlMode.inpaint), { Path( - "models/controlnet/FLUX.1-dev-Controlnet-Inpainting-Alpha.safetensors" - ): "https://huggingface.co/alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Alpha/resolve/main/diffusion_pytorch_model.safetensors" + "models/controlnet/FLUX.1-dev-Controlnet-Inpainting-Beta.safetensors" + ): "https://huggingface.co/alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta/resolve/main/diffusion_pytorch_model.safetensors" }, ), ModelResource( @@ -693,6 +693,18 @@ def __hash__(self): ): "https://huggingface.co/XLabs-AI/flux-controlnet-collections/resolve/main/flux-depth-controlnet-v3.safetensors" }, ), + ModelResource( + "Reference (Flux)", + ResourceId(ResourceKind.ip_adapter, Arch.flux, ControlMode.reference), + { + Path( + "models/clip_vision/sigclip_vision_patch14_384.safetensors" + ): "https://huggingface.co/Comfy-Org/sigclip_vision_384/resolve/main/sigclip_vision_patch14_384.safetensors", + Path( + "models/style_models/flux1-redux-dev.safetensors" + ): "https://files.interstice.cloud/models/flux1-redux-dev.safetensors", + }, + ), ] prefetch_models = [