Update resources

Acly · Nov 28, 2024 · fbe3781 · fbe3781
1 parent 7497b75
commit fbe3781
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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 = [