set face restore upscale to 1; realesrgan till to 512

This commit is contained in:
Qing 2023-04-03 13:19:26 +08:00
parent ed798bb47c
commit dd1d45aa79
3 changed files with 3 additions and 3 deletions

View File

@ -25,7 +25,7 @@ class GFPGANPlugin(BasePlugin):
# Use GFPGAN for face enhancement
self.face_enhancer = MyGFPGANer(
model_path=model_path,
upscale=2,
upscale=1,
arch="clean",
channel_multiplier=2,
device=device,

View File

@ -70,7 +70,7 @@ class RealESRGANUpscaler(BasePlugin):
model_path=model_path,
model=model_info["model"](),
half=True if "cuda" in str(device) else False,
tile=640,
tile=512,
tile_pad=10,
pre_pad=10,
device=device,

View File

@ -24,7 +24,7 @@ class RestoreFormerPlugin(BasePlugin):
self.face_enhancer = MyGFPGANer(
model_path=model_path,
upscale=2,
upscale=1,
arch="RestoreFormer",
channel_multiplier=2,
device=device,