CheckSim commited on
Commit
5f76bf1
·
verified ·
1 Parent(s): 3ef6ae3

Upload 2 files

Browse files
pipeline_stable_diffusion_xl_instantid.py CHANGED
@@ -540,6 +540,7 @@ class StableDiffusionXLInstantIDPipeline(StableDiffusionXLControlNetPipeline):
540
  else:
541
  prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
542
 
 
543
  prompt_image_emb = self.image_proj_model(prompt_image_emb)
544
  return prompt_image_emb
545
 
 
540
  else:
541
  prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
542
 
543
+ self.image_proj_model.to(device=device, dtype=dtype)
544
  prompt_image_emb = self.image_proj_model(prompt_image_emb)
545
  return prompt_image_emb
546
 
requirements.txt CHANGED
@@ -8,3 +8,4 @@ insightface
8
  onnxruntime
9
  opencv-python
10
  huggingface-hub
 
 
8
  onnxruntime
9
  opencv-python
10
  huggingface-hub
11
+ peft