all generators use get_generator() now

2024-02-15 12:58:57 +03:00 · 2024-02-15 12:58:57 +03:00 · ab3acb20df
parent abd3f860d1
commit ab3acb20df
1 changed files with 1 additions and 1 deletions
--- a/ip_adapter/ip_adapter.py
+++ b/ip_adapter/ip_adapter.py
@ -268,7 +268,7 @@ class IPAdapterXL(IPAdapter):
            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)

-        self.generator = torch.Generator(self.device).manual_seed(seed) if seed is not None else None
+        self.generator = get_generator(seed, self.device)
        
        images = self.pipe(
            prompt_embeds=prompt_embeds,