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

Verify architecture for loaded Spandrel models

Browse source
This commit is contained in:
Aarni Koskela 2023-12-30 16:37:03 +02:00
parent c756133541
commit 4ad0c0c0a8
8 changed files with 22 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/codeformer_model.py
View file

@ -37,6 +37,7 @@ class FaceRestorerCodeFormer(face_restoration_utils.CommonFaceRestoration):
return modelloader.load_spandrel_model(
model_path,
device=devices.device_codeformer,
expected_architecture='CodeFormer',
).model
raise ValueError("No codeformer model found")