Update app.py

app.py

@@ -114,7 +114,7 @@ def scheduler_swap_callback(pipeline, step_index, timestep, callback_kwargs):
      #   pipe.vae = vae_a
      #   pipe.unet = unet_a
         torch.backends.cudnn.deterministic = False
-        pipe.unet.set_default_attn_processor()
+        #pipe.unet.set_default_attn_processor()
         print("-- swapping scheduler --")
       #  pipeline.scheduler = heun_scheduler
         #pipe.scheduler.set_timesteps(num_inference_steps*.70)
@@ -248,7 +248,7 @@ def upload_to_ftp(filename):
 def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
     filename= f'rv_C_{timestamp}.txt'
     with open(filename, "w") as f:
-        f.write(f"Realvis 5.0 (Tester C) \n")
+        f.write(f"Realvis 5.0 (Tester D) \n")
         f.write(f"Date/time: {timestamp} \n")
         f.write(f"Prompt: {prompt} \n")
         f.write(f"Steps: {num_inference_steps} \n")
@@ -300,14 +300,14 @@ def generate_30(
     #uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     batch_options = options.copy()
     rv_image = pipe(**batch_options).images[0]
-    sd_image_path = f"rv_C_{timestamp}.png"
+    sd_image_path = f"rv_D_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(sd_image_path)    
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)
     downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
-    downscale_path = f"rv50_upscale_{timestamp}.png"
+    downscale_path = f"rv_D_upscale_{timestamp}.png"
     downscale1.save(downscale_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(downscale_path) 
     unique_name = str(uuid.uuid4()) + ".png"  
@@ -348,7 +348,7 @@ def generate_60(
     uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     batch_options = options.copy()
     rv_image = pipe(**batch_options).images[0]
-    sd_image_path = f"rv_C_{timestamp}.png"
+    sd_image_path = f"rv_D_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)    
     unique_name = str(uuid.uuid4()) + ".png"  
@@ -389,7 +389,7 @@ def generate_90(
     uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     batch_options = options.copy()
     rv_image = pipe(**batch_options).images[0]
-    sd_image_path = f"rv_C_{timestamp}.png"
+    sd_image_path = f"rv_D_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)    
     unique_name = str(uuid.uuid4()) + ".png"