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

Fixed issue with token replacements

Browse Source
This commit is contained in:
Jaret Burkett
2023-09-07 13:42:39 -06:00
parent 3feb663a51
commit 92a086d5a5
3 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/embedding.py
View File

@@ -159,7 +159,7 @@ class Embedding:
output_prompt = output_prompt.replace(to_replace, replace_with)
# see how many times replace_with is in the prompt
num_instances = prompt.count(replace_with)
num_instances = output_prompt.count(replace_with)
if num_instances == 0 and add_if_not_present:
# add it to the beginning of the prompt

1
toolkit/stable_diffusion_model.py
View File

@@ -614,6 +614,7 @@ class StableDiffusion:
return latents
def encode_prompt(self, prompt, num_images_per_prompt=1) -> PromptEmbeds:
# sd1.5 embeddings are (bs, 77, 768)
prompt = prompt
# if it is not a list, make it one
if not isinstance(prompt, list):