model2 commited on
Commit
30fdf5a
·
1 Parent(s): 20dcbc9

model preloading fixes

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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], "patcher", None), dict)
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