StableDiffusion-3.5-Large-lora-test

Running on Zero

App Files Files Community

1inkusFace commited on Apr 14

Commit

38ea7b1

verified ·

1 Parent(s): 9c98320

Update app.py

Browse files

Files changed (1) hide show

app.py +6 -6

app.py CHANGED Viewed

@@ -145,8 +145,8 @@ def infer_60(
     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
-        upscale = upscaler_2(sd_image, tiling=True, tile_width=512, tile_height=512)
-        upscale2 = upscaler_2(upscale, tiling=True, tile_width=512, tile_height=512)
     print('-- got upscaled image --')
     downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
     downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)
@@ -199,8 +199,8 @@ def infer_90(
     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
-        upscale = upscaler_2(sd_image, tiling=True, tile_width=512, tile_height=512)
-        upscale2 = upscaler_2(upscale, tiling=True, tile_width=512, tile_height=512)
     print('-- got upscaled image --')
     downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
     downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)
@@ -253,8 +253,8 @@ def infer_110(
     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
-        upscale = upscaler_2(sd_image, tiling=True, tile_width=512, tile_height=512)
-        upscale2 = upscaler_2(upscale, tiling=True, tile_width=512, tile_height=512)
     print('-- got upscaled image --')
     downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
     downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)

     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
+        upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
+        upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
     print('-- got upscaled image --')
     downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
     downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)
     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
+        upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
+        upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
     print('-- got upscaled image --')
     downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
     downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)
     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
+        upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
+        upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
     print('-- got upscaled image --')
     downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
     downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)