jayparmr commited on
Commit
c95142c
1 Parent(s): fe95d1e

Upload folder using huggingface_hub

Browse files
external/scripts/day_night_ip2p.py CHANGED
@@ -28,7 +28,7 @@ class Script:
28
 
29
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
30
  model_id,
31
- use_auth_token=get_hf_token(),
32
  torch_dtype=torch.float16,
33
  safety_checker=None,
34
  ).to("cuda")
 
28
 
29
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
30
  model_id,
31
+ token=get_hf_token(),
32
  torch_dtype=torch.float16,
33
  safety_checker=None,
34
  ).to("cuda")
internals/pipelines/commons.py CHANGED
@@ -44,7 +44,7 @@ class Text2Img(AbstractPipeline):
44
  pipe = StableDiffusionXLPipeline.from_pretrained(
45
  model_dir,
46
  torch_dtype=torch.float16,
47
- use_auth_token=get_hf_token(),
48
  use_safetensors=True,
49
  variant=get_base_model_variant(),
50
  )
@@ -53,7 +53,7 @@ class Text2Img(AbstractPipeline):
53
  self.pipe = pipe
54
  else:
55
  self.pipe = two_step_pipeline.from_pretrained(
56
- model_dir, torch_dtype=torch.float16, use_auth_token=get_hf_token()
57
  ).to("cuda")
58
  self.__patch()
59
 
@@ -143,13 +143,13 @@ class Img2Img(AbstractPipeline):
143
  self.pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
144
  model_dir,
145
  torch_dtype=torch.float16,
146
- use_auth_token=get_hf_token(),
147
  variant=get_base_model_variant(),
148
  use_safetensors=True,
149
  ).to("cuda")
150
  else:
151
  self.pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
152
- model_dir, torch_dtype=torch.float16, use_auth_token=get_hf_token()
153
  ).to("cuda")
154
  self.__patch()
155
 
 
44
  pipe = StableDiffusionXLPipeline.from_pretrained(
45
  model_dir,
46
  torch_dtype=torch.float16,
47
+ token=get_hf_token(),
48
  use_safetensors=True,
49
  variant=get_base_model_variant(),
50
  )
 
53
  self.pipe = pipe
54
  else:
55
  self.pipe = two_step_pipeline.from_pretrained(
56
+ model_dir, torch_dtype=torch.float16, token=get_hf_token()
57
  ).to("cuda")
58
  self.__patch()
59
 
 
143
  self.pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
144
  model_dir,
145
  torch_dtype=torch.float16,
146
+ token=get_hf_token(),
147
  variant=get_base_model_variant(),
148
  use_safetensors=True,
149
  ).to("cuda")
150
  else:
151
  self.pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
152
+ model_dir, torch_dtype=torch.float16, token=get_hf_token()
153
  ).to("cuda")
154
  self.__patch()
155
 
internals/pipelines/controlnets.py CHANGED
@@ -67,7 +67,7 @@ class StableDiffusionNetworkModelPipelineLoader:
67
  kwargs = {
68
  "pretrained_model_name_or_path": get_model_dir(),
69
  "torch_dtype": torch.float16,
70
- "use_auth_token": get_hf_token(),
71
  "cache_dir": get_hf_cache_dir(),
72
  }
73
  else:
 
67
  kwargs = {
68
  "pretrained_model_name_or_path": get_model_dir(),
69
  "torch_dtype": torch.float16,
70
+ "token": get_hf_token(),
71
  "cache_dir": get_hf_cache_dir(),
72
  }
73
  else:
internals/pipelines/inpainter.py CHANGED
@@ -36,7 +36,7 @@ class InPainter(AbstractPipeline):
36
  get_inpaint_model_path(),
37
  torch_dtype=torch.float16,
38
  cache_dir=get_hf_cache_dir(),
39
- use_auth_token=get_hf_token(),
40
  variant=get_base_inpaint_model_variant(),
41
  ).to("cuda")
42
  else:
@@ -44,7 +44,7 @@ class InPainter(AbstractPipeline):
44
  get_inpaint_model_path(),
45
  torch_dtype=torch.float16,
46
  cache_dir=get_hf_cache_dir(),
47
- use_auth_token=get_hf_token(),
48
  ).to("cuda")
49
 
50
  disable_safety_checker(self.pipe)
 
36
  get_inpaint_model_path(),
37
  torch_dtype=torch.float16,
38
  cache_dir=get_hf_cache_dir(),
39
+ token=get_hf_token(),
40
  variant=get_base_inpaint_model_variant(),
41
  ).to("cuda")
42
  else:
 
44
  get_inpaint_model_path(),
45
  torch_dtype=torch.float16,
46
  cache_dir=get_hf_cache_dir(),
47
+ token=get_hf_token(),
48
  ).to("cuda")
49
 
50
  disable_safety_checker(self.pipe)
internals/pipelines/replace_background.py CHANGED
@@ -59,7 +59,7 @@ class ReplaceBackground(AbstractPipeline):
59
  controlnet=controlnet_model,
60
  torch_dtype=torch.float16,
61
  cache_dir=get_hf_cache_dir(),
62
- use_auth_token=get_hf_token(),
63
  )
64
  pipe.enable_xformers_memory_efficient_attention()
65
  pipe.enable_vae_slicing()
 
59
  controlnet=controlnet_model,
60
  torch_dtype=torch.float16,
61
  cache_dir=get_hf_cache_dir(),
62
+ token=get_hf_token(),
63
  )
64
  pipe.enable_xformers_memory_efficient_attention()
65
  pipe.enable_vae_slicing()