Unverified Commit bae91855 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #12737 from yajunzhng/master

tell RealESRGANer which device to run on, could be cuda, M1, or other…
parents 0232a987 f29b4cd7
Loading
Loading
Loading
Loading
+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]