model2 commited on
Commit
04dde6a
·
1 Parent(s): 30fdf5a

fix preloading

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,6 @@ def import_custom_nodes() -> None:
79
 
80
 
81
  # Preload nodes, models.
82
- import_custom_nodes()
83
  loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
84
  upscalemodelloader = NODE_CLASS_MAPPINGS["UpscaleModelLoader"]()
85
  reactorloadfacemodel = NODE_CLASS_MAPPINGS["ReActorLoadFaceModel"]()
@@ -215,6 +214,7 @@ add_extra_model_paths()
215
 
216
  @spaces.GPU(duration=60)
217
  def advance_blur(input_image):
 
218
  with torch.inference_mode():
219
  loaded_input_image = loadimage.load_image(
220
  image=input_image,
 
79
 
80
 
81
  # Preload nodes, models.
 
82
  loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
83
  upscalemodelloader = NODE_CLASS_MAPPINGS["UpscaleModelLoader"]()
84
  reactorloadfacemodel = NODE_CLASS_MAPPINGS["ReActorLoadFaceModel"]()
 
214
 
215
  @spaces.GPU(duration=60)
216
  def advance_blur(input_image):
217
+ import_custom_nodes()
218
  with torch.inference_mode():
219
  loaded_input_image = loadimage.load_image(
220
  image=input_image,