github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 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

2
toolkit/config_modules.py
View File

@@ -99,3 +99,5 @@ class SliderConfig:
anchors = [SliderConfigAnchors(**anchor) for anchor in anchors]
self.anchors: List[SliderConfigAnchors] = anchors
self.resolutions: List[List[int]] = kwargs.get('resolutions', [[512, 512]])
self.prompt_file: str = kwargs.get('prompt_file', '')
self.prompt_tensors: str = kwargs.get('prompt_tensors', '')

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