flux-style-shaping

Runtime error

Dreamspire commited on Jul 14

Commit

1fb62f2

1 Parent(s): 580d7fc

update load_models_gpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -311,7 +311,9 @@ model_loaders = [
     layerutility_imageblend_v2
 ]
 print("model_management.load_models_gpu(model_loaders)")
-model_management.load_models_gpu(model_loaders)
 print("model_management.load_models_gpu(model_loaders) done")

     layerutility_imageblend_v2
 ]
 print("model_management.load_models_gpu(model_loaders)")
+model_management.load_models_gpu([
+    loader[0].patcher if hasattr(loader[0], 'patcher') else loader[0] for loader in model_loaders
+])
 print("model_management.load_models_gpu(model_loaders) done")