hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
yajun
2023-08-23 14:31:38 +08:00
parent 68f336bd99
commit f29b4cd7cb
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/realesrgan_model.py
View File

@@ -55,6 +55,7 @@ class UpscalerRealESRGAN(Upscaler):
half=not cmd_opts.no_half and not cmd_opts.upcast_sampling, half=not cmd_opts.no_half and not cmd_opts.upcast_sampling,
tile=opts.ESRGAN_tile, tile=opts.ESRGAN_tile,
tile_pad=opts.ESRGAN_tile_overlap, tile_pad=opts.ESRGAN_tile_overlap,
device=self.device,
) )
upsampled = upsampler.enhance(np.array(img), outscale=info.scale)[0] upsampled = upsampler.enhance(np.array(img), outscale=info.scale)[0]