model2 commited on
Commit
860ae1a
·
1 Parent(s): 02e495f

fix preloading

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -91,13 +91,13 @@ reactorfaceswap = NODE_CLASS_MAPPINGS["ReActorFaceSwap"]()
91
  imageupscalewithmodel = NODE_CLASS_MAPPINGS["ImageUpscaleWithModel"]()
92
  saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
93
  UPSCALE_MODEL = upscalemodelloader.load_model(model_name="ESRGAN/4x_NMKD-Siax_200k.pth")
94
- model_loaders = [
95
- UPSCALE_MODEL,
96
- FACE_MODEL,
97
- ]
98
- model_management.load_models_gpu([
99
- loader[0].patcher if hasattr(loader[0], 'patcher') else loader[0] for loader in model_loaders
100
- ])
101
 
102
 
103
  def load_extra_path_config(yaml_path):
 
91
  imageupscalewithmodel = NODE_CLASS_MAPPINGS["ImageUpscaleWithModel"]()
92
  saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
93
  UPSCALE_MODEL = upscalemodelloader.load_model(model_name="ESRGAN/4x_NMKD-Siax_200k.pth")
94
+ # model_loaders = [
95
+ # UPSCALE_MODEL,
96
+ # FACE_MODEL,
97
+ # ]
98
+ # model_management.load_models_gpu([
99
+ # loader[0].patcher if hasattr(loader[0], 'patcher') else loader[0] for loader in model_loaders
100
+ # ])
101
 
102
 
103
  def load_extra_path_config(yaml_path):