Update app.py

app.py

@@ -19,7 +19,7 @@ import paramiko
 import datetime
 import cyper
 from image_gen_aux import UpscaleWithModel
-
+'''
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
 torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
@@ -29,10 +29,10 @@ torch.backends.cudnn.benchmark = False
 # torch.backends.cuda.preferred_blas_library="cublas"
 # torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
-
+'''
 
 DESCRIPTIONXX = """
-    ## ⚡⚡⚡⚡ REALVISXL V5.0 BF16 (Tester C) ⚡⚡⚡⚡
+    ## ⚡⚡⚡⚡ REALVISXL V5.0 BF16 (Tester D) ⚡⚡⚡⚡
 """
 
 examples = [
@@ -141,7 +141,7 @@ def load_and_prepare_model():
    # pipe.to(device=device, dtype=torch.bfloat16)
     pipe.to(device)
     pipe.to(torch.bfloat16)
-    pipe.vae.set_default_attn_processor()
+    #pipe.vae.set_default_attn_processor()
     print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
     pipe.watermark=None
     pipe.safety_checker=None  
@@ -277,7 +277,7 @@ def generate_30(
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     seed = random.randint(0, MAX_SEED)
-  #  generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cuda').manual_seed(seed)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],
@@ -286,7 +286,7 @@ def generate_30(
         "height": height,
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
-   #     "generator": generator,
+        "generator": generator,
         "output_type": "pil",
         "callback_on_step_end": pyx.scheduler_swap_callback
     }
@@ -328,7 +328,7 @@ def generate_60(
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     seed = random.randint(0, MAX_SEED)
-  #  generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cuda').manual_seed(seed)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],
@@ -337,7 +337,7 @@ def generate_60(
         "height": height,
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
-  #      "generator": generator,
+        "generator": generator,
         "output_type": "pil",
         "callback_on_step_end": scheduler_swap_callback
     }
@@ -369,7 +369,7 @@ def generate_90(
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     seed = random.randint(0, MAX_SEED)
-  #  generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cuda').manual_seed(seed)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],
@@ -378,7 +378,7 @@ def generate_90(
         "height": height,
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
- #       "generator": generator,
+        "generator": generator,
         "output_type": "pil",
         "callback_on_step_end": scheduler_swap_callback
     }
@@ -561,7 +561,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
         outputs=[result],
     )
 
-    gr.Markdown("### REALVISXL V5.0")
+    gr.Markdown("### REALVISXL V5.0 Default Mode")
     predefined_gallery = gr.Gallery(label="REALVISXL V5.0", columns=3, show_label=False, value=load_predefined_images1())
 
     #gr.Markdown("### LIGHTNING V5.0")