Bugfix for offline mode with local models in images modules

Woolverine94 · Jun 25, 2024 · 46f9835 · 46f9835
1 parent f0be01d
commit 46f9835
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 9 deletions.
diff --git a/ressources/magicmix.py b/ressources/magicmix.py
@@ -84,6 +84,7 @@ def image_magicmix(
             torch_dtype=model_arch,
             use_safetensors=True, 
             load_safety_checker=False if (nsfw_filter_final == None) else True,
+            local_files_only=True if offline_test() else None,
 #            safety_checker=nsfw_filter_final, 
 #            feature_extractor=feat_ex
         )
@@ -98,7 +99,7 @@ def image_magicmix(
             safety_checker=nsfw_filter_final, 
             feature_extractor=feat_ex,
             resume_download=True,
-            local_files_only=True if offline_test() else None
+            local_files_only=True if offline_test() else None,
         )
 
     pipe_magicmix = schedulerer(pipe_magicmix, sampler_magicmix)

diff --git a/ressources/paintbyex.py b/ressources/paintbyex.py
@@ -82,6 +82,7 @@ def image_paintbyex(
             torch_dtype=model_arch,
 #            use_safetensors=True, 
             load_safety_checker=False if (nsfw_filter_final == None) else True,
+            local_files_only=True if offline_test() else None,
 #            safety_checker=nsfw_filter_final, 
 #            feature_extractor=feat_ex, 
         )

diff --git a/ressources/txt2img_kd.py b/ressources/txt2img_kd.py
@@ -90,6 +90,7 @@ def image_txt2img_kd(
                 modelid_txt2img_kd,
                 torch_dtype=model_arch,
                 use_safetensors=True,
+                local_files_only=True if offline_test() else None,
             )
         else :
             pipe_txt2img_kd = AutoPipelineForText2Image.from_pretrained(
@@ -99,14 +100,15 @@ def image_txt2img_kd(
                 variant="fp16",
                 use_safetensors=True,
                 resume_download=True,
-                local_files_only=True if offline_test() else None
+                local_files_only=True if offline_test() else None,
             )
     else :
         if modelid_txt2img_kd[0:9] == "./models/" :
             pipe_txt2img_kd = AutoPipelineForText2Image.from_single_file(
                 modelid_txt2img_kd,
                 torch_dtype=model_arch,
                 use_safetensors=True,
+                local_files_only=True if offline_test() else None,
             )
         else :
             pipe_txt2img_kd = AutoPipelineForText2Image.from_pretrained(
@@ -115,7 +117,7 @@ def image_txt2img_kd(
                 torch_dtype=model_arch,
                 use_safetensors=True,
                 resume_download=True,
-                local_files_only=True if offline_test() else None
+                local_files_only=True if offline_test() else None,
             )
 
     pipe_txt2img_kd = schedulerer(pipe_txt2img_kd, sampler_txt2img_kd)

diff --git a/ressources/txt2img_mjm.py b/ressources/txt2img_mjm.py
@@ -98,6 +98,7 @@ def image_txt2img_mjm(
             torch_dtype=model_arch, 
 #            use_safetensors=True, 
             load_safety_checker=False if (nsfw_filter_final == None) else True,
+            local_files_only=True if offline_test() else None,
 #            safety_checker=nsfw_filter_final, 
 #            feature_extractor=feat_ex,
         )
@@ -110,7 +111,7 @@ def image_txt2img_mjm(
             safety_checker=nsfw_filter_final, 
             feature_extractor=feat_ex,
             resume_download=True,
-            local_files_only=True if offline_test() else None
+            local_files_only=True if offline_test() else None,
         )
 
     pipe_txt2img_mjm = schedulerer(pipe_txt2img_mjm, sampler_txt2img_mjm)

diff --git a/ressources/txt2img_paa.py b/ressources/txt2img_paa.py
@@ -104,6 +104,7 @@ def image_txt2img_paa(
                 torch_dtype=model_arch,
                 use_safetensors=True, 
                 load_safety_checker=False if (nsfw_filter_final == None) else True,
+                local_files_only=True if offline_test() else None,
     #            safety_checker=nsfw_filter_final, 
     #            feature_extractor=feat_ex,
             )
@@ -117,7 +118,7 @@ def image_txt2img_paa(
                 safety_checker=nsfw_filter_final, 
                 feature_extractor=feat_ex,
                 resume_download=True,
-                local_files_only=True if offline_test() else None
+                local_files_only=True if offline_test() else None,
             )
         pipe_txt2img_paa.scheduler = LCMScheduler.from_config(pipe_txt2img_paa.scheduler.config)
         pipe_txt2img_paa.scheduler.config.prediction_type = "v_prediction"
@@ -133,7 +134,7 @@ def image_txt2img_paa(
             torch_dtype=model_arch,
             use_safetensors=True,
             resume_download=True,
-            local_files_only=True if offline_test() else None
+            local_files_only=True if offline_test() else None,
         )
 
         transformer_txt2img_paa = PeftModel.from_pretrained(
@@ -148,6 +149,7 @@ def image_txt2img_paa(
                 torch_dtype=model_arch,
                 use_safetensors=True,
                 load_safety_checker=False if (nsfw_filter_final == None) else True,
+                local_files_only=True if offline_test() else None,
 #                safety_checker=nsfw_filter_final,
 #                feature_extractor=feat_ex,
             )
@@ -161,7 +163,7 @@ def image_txt2img_paa(
                 safety_checker=nsfw_filter_final,
                 feature_extractor=feat_ex,
                 resume_download=True,
-                local_files_only=True if offline_test() else None
+                local_files_only=True if offline_test() else None,
             )
         pipe_txt2img_paa.scheduler = LCMScheduler.from_config(pipe_txt2img_paa.scheduler.config)
         pipe_txt2img_paa.scheduler.config.timestep_spacing = "trailing"
@@ -173,6 +175,7 @@ def image_txt2img_paa(
                 torch_dtype=model_arch,
                 use_safetensors=True,
                 load_safety_checker=False if (nsfw_filter_final == None) else True,
+                local_files_only=True if offline_test() else None,
     #            safety_checker=nsfw_filter_final,
     #            feature_extractor=feat_ex,
             )
@@ -185,7 +188,7 @@ def image_txt2img_paa(
                 safety_checker=nsfw_filter_final,
                 feature_extractor=feat_ex,
                 resume_download=True,
-                local_files_only=True if offline_test() else None
+                local_files_only=True if offline_test() else None,
             )
         pipe_txt2img_paa = schedulerer(pipe_txt2img_paa, sampler_txt2img_paa)
 
@@ -196,6 +199,7 @@ def image_txt2img_paa(
                 torch_dtype=model_arch,
                 use_safetensors=True, 
                 load_safety_checker=False if (nsfw_filter_final == None) else True,
+                local_files_only=True if offline_test() else None,
     #            safety_checker=nsfw_filter_final, 
     #            feature_extractor=feat_ex,
             )
@@ -208,7 +212,7 @@ def image_txt2img_paa(
                 safety_checker=nsfw_filter_final, 
                 feature_extractor=feat_ex,
                 resume_download=True,
-                local_files_only=True if offline_test() else None
+                local_files_only=True if offline_test() else None,
             )
         pipe_txt2img_paa = schedulerer(pipe_txt2img_paa, sampler_txt2img_paa)