github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Bug fixes, work on maing IP adapters more customizable.

Browse Source
This commit is contained in:
Jaret Burkett
2023-12-24 08:32:39 -07:00
parent 7703e3a15e
commit 0f8daa5612
7 changed files with 243 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
toolkit/prompt_utils.py
View File

@@ -39,10 +39,11 @@ class PromptEmbeds:
return self
def detach(self):
self.text_embeds = self.text_embeds.detach()
if self.pooled_embeds is not None:
self.pooled_embeds = self.pooled_embeds.detach()
return self
new_embeds = self.clone()
new_embeds.text_embeds = new_embeds.text_embeds.detach()
if new_embeds.pooled_embeds is not None:
new_embeds.pooled_embeds = new_embeds.pooled_embeds.detach()
return new_embeds
def clone(self):
if self.pooled_embeds is not None: