ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-13 22:49:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP diffusers pipeline is weird. Starting to hate sdxl

Browse Source
This commit is contained in:
Jaret Burkett
2023-07-27 17:35:24 -06:00
parent b2e2e4bf47
commit c7640b0865
6 changed files with 670 additions and 233 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
toolkit/stable_diffusion_model.py
View File

@@ -30,10 +30,10 @@ class PromptEmbeds:
self.text_embeds = args
self.pooled_embeds = None
def to(self, **kwargs):
self.text_embeds = self.text_embeds.to(**kwargs)
def to(self, *args, **kwargs):
self.text_embeds = self.text_embeds.to(*args, **kwargs)
if self.pooled_embeds is not None:
self.pooled_embeds = self.pooled_embeds.to(**kwargs)
self.pooled_embeds = self.pooled_embeds.to(*args, **kwargs)
return self