mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-15 14:50:55 -08:00
Code Issues Projects Releases Packages Wiki Activity

When device is MPS, use CPU for GFPGAN instead

Browse source 
GFPGAN will not work if the device is MPS, so default to CPU instead.
This commit is contained in:
brkirch 2022-09-30 23:53:25 -04:00
parent 84e97a98c5
commit bdaa36c844
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
modules/gfpgan_model.py
View file

@ -21,7 +21,7 @@ def gfpgann():
global loaded_gfpgan_model
global model_path
if loaded_gfpgan_model is not None:
loaded_gfpgan_model.gfpgan.to(shared.device)
loaded_gfpgan_model.gfpgan.to(devices.device_gfpgan)
return loaded_gfpgan_model
if gfpgan_constructor is None:
@ -36,8 +36,8 @@ def gfpgann():
else:
print("Unable to load gfpgan model!")
return None
model = gfpgan_constructor(model_path=model_file, upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
model.gfpgan.to(shared.device)
model = gfpgan_constructor(model_path=model_file, upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None, device=devices.device_gfpgan)
model.gfpgan.to(devices.device_gfpgan)
loaded_gfpgan_model = model
return model