yanze commited on
Commit
b6d5b93
1 Parent(s): dd561ba

Update pulid/pipeline.py

Browse files
Files changed (1) hide show
  1. pulid/pipeline.py +1 -0
pulid/pipeline.py CHANGED
@@ -217,6 +217,7 @@ class PuLIDPipeline:
217
  return torch.cat((uncond_id_embedding, id_embedding), dim=0)
218
 
219
  def inference(self, prompt, size, prompt_n='', image_embedding=None, id_scale=1.0, guidance_scale=1.2, steps=4):
 
220
  images = self.pipe(
221
  prompt=prompt,
222
  negative_prompt=prompt_n,
 
217
  return torch.cat((uncond_id_embedding, id_embedding), dim=0)
218
 
219
  def inference(self, prompt, size, prompt_n='', image_embedding=None, id_scale=1.0, guidance_scale=1.2, steps=4):
220
+ print(image_embedding.dtype)
221
  images = self.pipe(
222
  prompt=prompt,
223
  negative_prompt=prompt_n,