Upload folder using huggingface_hub

external/scripts/day_night_ip2p.py

@@ -28,7 +28,7 @@ class Script:
 
         pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
             model_id,
-            use_auth_token=get_hf_token(),
+            token=get_hf_token(),
             torch_dtype=torch.float16,
             safety_checker=None,
         ).to("cuda")

internals/pipelines/commons.py

@@ -44,7 +44,7 @@ class Text2Img(AbstractPipeline):
             pipe = StableDiffusionXLPipeline.from_pretrained(
                 model_dir,
                 torch_dtype=torch.float16,
-                use_auth_token=get_hf_token(),
+                token=get_hf_token(),
                 use_safetensors=True,
                 variant=get_base_model_variant(),
             )
@@ -53,7 +53,7 @@ class Text2Img(AbstractPipeline):
             self.pipe = pipe
         else:
             self.pipe = two_step_pipeline.from_pretrained(
-                model_dir, torch_dtype=torch.float16, use_auth_token=get_hf_token()
+                model_dir, torch_dtype=torch.float16, token=get_hf_token()
             ).to("cuda")
         self.__patch()
 
@@ -143,13 +143,13 @@ class Img2Img(AbstractPipeline):
             self.pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
                 model_dir,
                 torch_dtype=torch.float16,
-                use_auth_token=get_hf_token(),
+                token=get_hf_token(),
                 variant=get_base_model_variant(),
                 use_safetensors=True,
             ).to("cuda")
         else:
             self.pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
-                model_dir, torch_dtype=torch.float16, use_auth_token=get_hf_token()
+                model_dir, torch_dtype=torch.float16, token=get_hf_token()
             ).to("cuda")
         self.__patch()
 

internals/pipelines/controlnets.py

@@ -67,7 +67,7 @@ class StableDiffusionNetworkModelPipelineLoader:
             kwargs = {
                 "pretrained_model_name_or_path": get_model_dir(),
                 "torch_dtype": torch.float16,
-                "use_auth_token": get_hf_token(),
+                "token": get_hf_token(),
                 "cache_dir": get_hf_cache_dir(),
             }
         else:

internals/pipelines/inpainter.py

@@ -36,7 +36,7 @@ class InPainter(AbstractPipeline):
                 get_inpaint_model_path(),
                 torch_dtype=torch.float16,
                 cache_dir=get_hf_cache_dir(),
-                use_auth_token=get_hf_token(),
+                token=get_hf_token(),
                 variant=get_base_inpaint_model_variant(),
             ).to("cuda")
         else:
@@ -44,7 +44,7 @@ class InPainter(AbstractPipeline):
                 get_inpaint_model_path(),
                 torch_dtype=torch.float16,
                 cache_dir=get_hf_cache_dir(),
-                use_auth_token=get_hf_token(),
+                token=get_hf_token(),
             ).to("cuda")
 
         disable_safety_checker(self.pipe)

internals/pipelines/replace_background.py

@@ -59,7 +59,7 @@ class ReplaceBackground(AbstractPipeline):
                 controlnet=controlnet_model,
                 torch_dtype=torch.float16,
                 cache_dir=get_hf_cache_dir(),
-                use_auth_token=get_hf_token(),
+                token=get_hf_token(),
             )
         pipe.enable_xformers_memory_efficient_attention()
         pipe.enable_vae_slicing()