RealVis_v5.0_BF16_IP_B

Running on Zero

ford442 commited on Jan 17

Commit

6b7a148

verified ·

1 Parent(s): c96270a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -275,7 +275,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)
@@ -365,7 +365,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)
@@ -456,7 +456,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.set_float32_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.set_float32_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.set_float32_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)