Update image_generator.py

image_generator.py

@@ -164,7 +164,7 @@ def generate_image_from_embeddings(
         loss_selection, additional_prompt):
     height = 512
     width = 512
-    num_inference_steps = 20
+    num_inference_steps = 50
     guidance_scale = 8
     generator = torch.manual_seed(seed)
     batch_size = 1
@@ -221,7 +221,7 @@ def generate_image_from_embeddings(
         )
 
         #### ADDITIONAL GUIDANCE ###
-        if i % 3 == 0:
+        if i % 5 == 0:
             # Requires grad on the latents
             latents = latents.detach().requires_grad_()
 
@@ -358,7 +358,6 @@ def generate_image_per_prompt_style(text_in, style_in,
     style_file = STYLE_EMBEDDINGS[style_in]
     print(f"style_file: {style_file}")
 
-    prompt = text_in
 
     style_seed = STYLE_SEEDS[idx]
 
@@ -370,6 +369,8 @@ def generate_image_per_prompt_style(text_in, style_in,
     embedding = load_embedding_bin(file_path)
     style_key = f"<{style_key}>"
 
+    prompt = f"{text_in} {style_key}"
+
     gen_style_image = generate_image_per_style(prompt, embedding, style_seed, style_key)
 
     gen_loss_image = generate_image_per_loss(prompt, embedding, style_seed, style_key, loss, additional_prompt)