yamildiego commited on
Commit
9c05a72
·
1 Parent(s): 2b2e3f7

last test

Browse files
{.checkpoints → checkpoints}/ControlNetModel/config.json RENAMED
File without changes
{.checkpoints → checkpoints}/ControlNetModel/diffusion_pytorch_model.safetensors RENAMED
File without changes
{.checkpoints → checkpoints}/ip-adapter.bin RENAMED
File without changes
handler.py CHANGED
@@ -57,7 +57,7 @@ class EndpointHandler():
57
  # ])
58
 
59
  self.controlnet_identitynet = ControlNetModel.from_pretrained(
60
- controlnet_path, torch_dtype=dtype
61
  )
62
 
63
  pretrained_model_name_or_path = "wangqixun/YamerMIX_v8"
 
57
  # ])
58
 
59
  self.controlnet_identitynet = ControlNetModel.from_pretrained(
60
+ controlnet_path, torch_dtype=dtype, local_files_only=True
61
  )
62
 
63
  pretrained_model_name_or_path = "wangqixun/YamerMIX_v8"