Bump Spandrel to 0.3.4; add spandrel-extra-arches for CodeFormer

This commit is contained in:
Aarni Koskela
2024-07-04 08:43:48 +03:00
parent a30b19dd55
commit f8fb74b93a
3 changed files with 32 additions and 7 deletions

View File

@@ -36,13 +36,11 @@ class FaceRestorerGFPGAN(face_restoration_utils.CommonFaceRestoration):
ext_filter=['.pth'],
):
if 'GFPGAN' in os.path.basename(model_path):
model = modelloader.load_spandrel_model(
return modelloader.load_spandrel_model(
model_path,
device=self.get_device(),
expected_architecture='GFPGAN',
).model
model.different_w = True # see https://github.com/chaiNNer-org/spandrel/pull/81
return model
raise ValueError("No GFPGAN model found")
def restore(self, np_image):