RealVis_v5.0_BF16_IP

Running on Zero

ford442 commited on Dec 31, 2024

Commit

3c13853

verified ·

1 Parent(s): 17a5ba3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -302,7 +302,7 @@ def generate_30(
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
-        torch.setfloat32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
@@ -355,7 +355,7 @@ def generate_60(
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
-        torch.setfloat32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
@@ -408,7 +408,7 @@ def generate_90(
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
-        torch.setfloat32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)

         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
+        #torch.setfloat32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
+        #torch.setfloat32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
+        #torch.setfloat32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)