perf: make upscaler use fp16 for better efficiency

5 months ago · 7100d3f9ea
parent 4fcfc363af
commit 7100d3f9ea
1 changed files with 1 additions and 0 deletions
--- 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)