From bc4adfbf2f83529ea6fd9998ee6a361a9120ff1b Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sat, 31 Aug 2024 09:40:18 +0800
Subject: [PATCH] Update ip_adapter.py

resume_download error
---
 foleycrafter/models/auffusion/loaders/ip_adapter.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/foleycrafter/models/auffusion/loaders/ip_adapter.py b/foleycrafter/models/auffusion/loaders/ip_adapter.py
index a081058..53b5e1f 100644
--- a/foleycrafter/models/auffusion/loaders/ip_adapter.py
+++ b/foleycrafter/models/auffusion/loaders/ip_adapter.py
@@ -134,7 +134,7 @@ class IPAdapterMixin:
         # Load the main state dict first.
         cache_dir = kwargs.pop("cache_dir", None)
         force_download = kwargs.pop("force_download", False)
-        resume_download = kwargs.pop("resume_download", False)
+        #resume_download = kwargs.pop("resume_download", False)
         proxies = kwargs.pop("proxies", None)
         local_files_only = kwargs.pop("local_files_only", None)
         token = kwargs.pop("token", None)
@@ -170,14 +170,13 @@ class IPAdapterMixin:
                     weights_name=weight_name,
                     cache_dir=cache_dir,
                     force_download=force_download,
-                    resume_download=resume_download,
                     proxies=proxies,
                     local_files_only=local_files_only,
                     token=token,
                     revision=revision,
                     subfolder=subfolder,
                     user_agent=user_agent,
-                )
+                )#resume_download=resume_download,
                 if weight_name.endswith(".safetensors"):
                     state_dict = {"image_proj": {}, "ip_adapter": {}}
                     with safe_open(model_file, framework="pt", device="cpu") as f:
@@ -372,7 +371,7 @@ class VPAdapterMixin:
         # Load the main state dict first.
         cache_dir = kwargs.pop("cache_dir", None)
         force_download = kwargs.pop("force_download", False)
-        resume_download = kwargs.pop("resume_download", False)
+        #resume_download = kwargs.pop("resume_download", False)
         proxies = kwargs.pop("proxies", None)
         local_files_only = kwargs.pop("local_files_only", None)
         token = kwargs.pop("token", None)
@@ -408,14 +407,13 @@ class VPAdapterMixin:
                     weights_name=weight_name,
                     cache_dir=cache_dir,
                     force_download=force_download,
-                    resume_download=resume_download,
                     proxies=proxies,
                     local_files_only=local_files_only,
                     token=token,
                     revision=revision,
                     subfolder=subfolder,
                     user_agent=user_agent,
-                )
+                )#resume_download=resume_download,
                 if weight_name.endswith(".safetensors"):
                     state_dict = {"image_proj": {}, "ip_adapter": {}}
                     with safe_open(model_file, framework="pt", device="cpu") as f: