Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make sde and ancestral schedulers reproducible

Browse Source
This commit is contained in:
Sergey Borisov 2023-07-14 05:25:09 +03:00
parent d8f88c09ea
commit 9fb0b0959f
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/invocations/sdxl.py
View File

@ -101,6 +101,11 @@ class SDXLTextToLatentsInvocation(BaseInvocation):
cross_attention_kwargs = None cross_attention_kwargs = None
with unet_info as unet: with unet_info as unet:
if "generator" in set(inspect.signature(scheduler.step).parameters.keys()):
extra_step_kwargs.update(
generator=torch.Generator(device=unet.device).manual_seed(0),
)
if not context.services.configuration.sequential_guidance: if not context.services.configuration.sequential_guidance:
prompt_embeds = torch.cat([negative_prompt_embeds, prompt_embeds], dim=0) prompt_embeds = torch.cat([negative_prompt_embeds, prompt_embeds], dim=0)
add_text_embeds = torch.cat([negative_pooled_prompt_embeds, pooled_prompt_embeds], dim=0) add_text_embeds = torch.cat([negative_pooled_prompt_embeds, pooled_prompt_embeds], dim=0)