Spaces:
Running
on
Zero
Running
on
Zero
Update pulid/pipeline.py
Browse files- 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,
|