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