Spaces:

multimodalart
/

flux-style-shaping

Running on Zero

multimodalart HF Staff commited on Dec 19, 2024

Commit

96f0f3d

verified ·

1 Parent(s): ec4e998

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -155,7 +155,7 @@ getimagesizeandcount = NODE_CLASS_MAPPINGS["GetImageSizeAndCount"]()
 depthanything_v2 = NODE_CLASS_MAPPINGS["DepthAnything_V2"]()
 imageresize = NODE_CLASS_MAPPINGS["ImageResize+"]()
-model_loaders = [CLIP_MODEL, VAE_MODEL, UNET_MODEL, CLIP_VISION_MODEL, DEPTH_MODEL, STYLE_MODEL]
 model_management.load_models_gpu([
     loader[0].patcher if hasattr(loader[0], 'patcher') else loader[0] for loader in model_loaders

 depthanything_v2 = NODE_CLASS_MAPPINGS["DepthAnything_V2"]()
 imageresize = NODE_CLASS_MAPPINGS["ImageResize+"]()
+model_loaders = [CLIP_MODEL, VAE_MODEL, UNET_MODEL, CLIP_VISION_MODEL]
 model_management.load_models_gpu([
     loader[0].patcher if hasattr(loader[0], 'patcher') else loader[0] for loader in model_loaders