Update app.py

app.py

@@ -147,7 +147,7 @@ models = {key: load_and_prepare_model(value) for key, value in MODEL_OPTIONS.ite
 
 MAX_SEED = np.iinfo(np.int32).max
 
-neg_prompt_2 = " 'unrealistic skin','unattractive face':1.3, 'low quality':1.1, ('dull color scheme', 'dull colors', 'digital noise':1.2),'amateurish', 'poorly drawn face':1.3, 'poorly drawn', 'distorted face' "
+neg_prompt_2 = " 'unrealistic skin','unattractive face':1.3, 'low quality':1.1, ('dull color scheme', 'dull colors', 'digital noise':1.2),'amateurish', 'poorly drawn face':1.3, 'poorly drawn', 'distorted face', 'low resolution', 'simplistic' "
 
 def upload_to_ftp(filename):
     try:
@@ -193,11 +193,11 @@ def generate_30(
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
-    prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
+    #prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
     options = {
         "prompt": [prompt] * num_images,
-        "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
-        "negative_prompt_2": [neg_prompt_2] * num_images if use_negative_prompt else None,
+        "negative_prompt": [negative_prompt],
+        "negative_prompt_2": [neg_prompt_2],
         "width": width,
         "height": height,
         "guidance_scale": guidance_scale,
@@ -222,8 +222,8 @@ def generate_30(
         f.write(f"Use Safetensors: no \n")
         f.write(f"Use Model Dtype: no \n")
         f.write(f"Model Scheduler: Euler_a custom before cuda \n")
-        f.write(f"Model VAE: stabilityai/sdxl-vae before cuda to bfloat after cuda \n")
-        f.write(f"Model UNET: stabilityai before cuda to bfloat after cuda \n")
+        f.write(f"Model VAE: stabilityai/sdxl-vae before cuda to bfloat with pipe \n")
+        f.write(f"Model UNET: stabilityai before cuda to bfloat with pipe \n")
     upload_to_ftp(filename) 
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
@@ -260,10 +260,11 @@ def generate_60(
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
-    prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
+    #prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
     options = {
         "prompt": [prompt] * num_images,
-        "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
+        "negative_prompt": [negative_prompt],
+        "negative_prompt_2": [neg_prompt_2],
         "width": width,
         "height": height,
         "guidance_scale": guidance_scale,
@@ -288,8 +289,8 @@ def generate_60(
         f.write(f"Use Safetensors: no \n")
         f.write(f"Use Model Dtype: no \n")
         f.write(f"Model Scheduler: Euler_a custom before cuda \n")
-        f.write(f"Model VAE: stabilityai/sdxl-vae before cuda to bfloat after cuda \n")
-        f.write(f"Model UNET: stabilityai before cuda to bfloat after cuda \n")
+        f.write(f"Model VAE: stabilityai/sdxl-vae before cuda to bfloat with pipe \n")
+        f.write(f"Model UNET: stabilityai before cuda to bfloat with pipe \n")
     upload_to_ftp(filename) 
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
@@ -326,10 +327,11 @@ def generate_90(
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
-    prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
+    #prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
     options = {
         "prompt": [prompt] * num_images,
-        "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
+        "negative_prompt": [negative_prompt],
+        "negative_prompt_2": [neg_prompt_2],
         "width": width,
         "height": height,
         "guidance_scale": guidance_scale,
@@ -354,8 +356,8 @@ def generate_90(
         f.write(f"Use Safetensors: no \n")
         f.write(f"Use Model Dtype: no \n")
         f.write(f"Model Scheduler: Euler_a custom before cuda \n")
-        f.write(f"Model VAE: stabilityai/sdxl-vae before cuda to bfloat after cuda \n")
-        f.write(f"Model UNET: stabilityai before cuda to bfloat after cuda \n")
+        f.write(f"Model VAE: stabilityai/sdxl-vae before cuda to bfloat with pipe \n")
+        f.write(f"Model UNET: stabilityai before cuda to bfloat with pipe \n")
     upload_to_ftp(filename) 
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()