pref: make upscaler use fp16 for better efficiency

brycedrennan · Jan 2, 2024 · b0d3a66 · b0d3a66
1 parent e2b3e82
commit b0d3a66
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/imaginairy/enhancers/upscale_realesrgan.py b/imaginairy/enhancers/upscale_realesrgan.py
@@ -31,6 +31,7 @@ def realesrgan_upsampler(tile=512, tile_pad=50, ultrasharp=False):
         tile=tile,
         device=device,
         tile_pad=tile_pad,
+        half=True
     )
 
     upsampler.device = torch.device(device)