1580158
/
IP-Adapter
mirror of https://github.com/tencent-ailab/IP-Adapter
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

seed list implemented for face id

pull/296/head
Ayça 2024-02-13 14:58:04 +03:00
parent 39b4dbf653
commit 9e82c1038a
1 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
ip_adapter/ip_adapter_faceid.py
View File

@ -397,7 +397,14 @@ class IPAdapterFaceIDPlus:
prompt_embeds = torch.cat([prompt_embeds_, image_prompt_embeds], dim=1)
negative_prompt_embeds = torch.cat([negative_prompt_embeds_, uncond_image_prompt_embeds], dim=1)
generator = torch.Generator(self.device).manual_seed(seed) if seed is not None else None
if seed is not None:
if isinstance(seed, list):
generator = [torch.Generator(self.device).manual_seed(seed_item) for seed_item in seed]
else:
generator = torch.Generator(self.device).manual_seed(seed)
else:
generator = None
images = self.pipe(
prompt_embeds=prompt_embeds,
negative_prompt_embeds=negative_prompt_embeds,
@ -461,7 +468,14 @@ class IPAdapterFaceIDXL(IPAdapterFaceID):
prompt_embeds = torch.cat([prompt_embeds, image_prompt_embeds], dim=1)
negative_prompt_embeds = torch.cat([negative_prompt_embeds, uncond_image_prompt_embeds], dim=1)
generator = torch.Generator(self.device).manual_seed(seed) if seed is not None else None
if seed is not None:
if isinstance(seed, list):
generator = [torch.Generator(self.device).manual_seed(seed_item) for seed_item in seed]
else:
generator = torch.Generator(self.device).manual_seed(seed)
else:
generator = None
images = self.pipe(
prompt_embeds=prompt_embeds,
negative_prompt_embeds=negative_prompt_embeds,
@ -530,7 +544,14 @@ class IPAdapterFaceIDPlusXL(IPAdapterFaceIDPlus):
prompt_embeds = torch.cat([prompt_embeds, image_prompt_embeds], dim=1)
negative_prompt_embeds = torch.cat([negative_prompt_embeds, uncond_image_prompt_embeds], dim=1)
generator = torch.Generator(self.device).manual_seed(seed) if seed is not None else None
if seed is not None:
if isinstance(seed, list):
generator = [torch.Generator(self.device).manual_seed(seed_item) for seed_item in seed]
else:
generator = torch.Generator(self.device).manual_seed(seed)
else:
generator = None
images = self.pipe(
prompt_embeds=prompt_embeds,
negative_prompt_embeds=negative_prompt_embeds,