Spaces:

mkthoma
/

StableDiffusion_with_VibranceLoss

Runtime error

mkthoma commited on Oct 13, 2023

Commit

1782f93

1 Parent(s): a443056

code update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -180,7 +180,7 @@ def generate_with_embs(text_embeddings, max_length, random_seed, loss_fn = None)
     return latents_to_pil(latents)[0]
 def generate_images(prompt, style_num=None, random_seed=41, custom_loss_fn = None):
-    eos_pos = get_EOS_pos_in_prompt(prompt)
     style_token_embedding = None
     if style_num:

     return latents_to_pil(latents)[0]
 def generate_images(prompt, style_num=None, random_seed=41, custom_loss_fn = None):
+    eos_pos = len(prompt.split())+1
     style_token_embedding = None
     if style_num: