Spaces:
Running
on
Zero
Running
on
Zero
model preloading fixes
Browse files
app.py
CHANGED
|
@@ -93,13 +93,14 @@ saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
|
|
| 93 |
upscale_model = upscalemodelloader.load_model(model_name="ESRGAN/4x_NMKD-Siax_200k.pth")
|
| 94 |
model_loaders = [
|
| 95 |
upscalemodelloader,
|
|
|
|
| 96 |
reactorfaceswap,
|
| 97 |
imageupscalewithmodel,
|
| 98 |
]
|
| 99 |
valid_models = [
|
| 100 |
getattr(loader[0], "patcher", loader[0])
|
| 101 |
for loader in model_loaders
|
| 102 |
-
if not isinstance(getattr(loader[0],
|
| 103 |
]
|
| 104 |
model_management.load_models_gpu(valid_models)
|
| 105 |
|
|
|
|
| 93 |
upscale_model = upscalemodelloader.load_model(model_name="ESRGAN/4x_NMKD-Siax_200k.pth")
|
| 94 |
model_loaders = [
|
| 95 |
upscalemodelloader,
|
| 96 |
+
reactorloadfacemodel,
|
| 97 |
reactorfaceswap,
|
| 98 |
imageupscalewithmodel,
|
| 99 |
]
|
| 100 |
valid_models = [
|
| 101 |
getattr(loader[0], "patcher", loader[0])
|
| 102 |
for loader in model_loaders
|
| 103 |
+
if not isinstance(loader[0], dict) and not isinstance(getattr(loader[0], 'patcher', None), dict)
|
| 104 |
]
|
| 105 |
model_management.load_models_gpu(valid_models)
|
| 106 |
|