Commit f29b4cd7 authored by yajun's avatar yajun
Browse files

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

parent 68f336bd

modules/realesrgan_model.py

+1 −0
Original line number Diff line number Diff line
@@ -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]

CRA Git | Maintained and supported by SUSTech CRA and CCSE