|
|
|
|
@ -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
|
|
|
|
|
|