tell RealESRGANer which device to run on, could be cuda, M1, or other GPU

2023-08-23 14:31:38 +08:00 · 2023-08-23 14:31:38 +08:00 · f29b4cd7cb
parent 68f336bd99
commit f29b4cd7cb
1 changed files with 1 additions and 0 deletions
--- a/modules/realesrgan_model.py
+++ b/modules/realesrgan_model.py
@ -55,6 +55,7 @@ class UpscalerRealESRGAN(Upscaler):
            half=not cmd_opts.no_half and not cmd_opts.upcast_sampling,
            tile=opts.ESRGAN_tile,
            tile_pad=opts.ESRGAN_tile_overlap,
+            device=self.device,
        )

        upsampled = upsampler.enhance(np.array(img), outscale=info.scale)[0]