Browse Source

gligen tuple

pull/559/head
BlenderNeko 2 years ago
parent
commit
0b07b2cc0f
  1. 9
      comfy/sample.py

9
comfy/sample.py

@ -44,11 +44,10 @@ def get_models_from_cond(cond, model_type):
def load_additional_models(positive, negative):
"""loads additional models in positive and negative conditioning"""
models = []
models += get_models_from_cond(positive, "control")
models += get_models_from_cond(negative, "control")
models += get_models_from_cond(positive, "gligen")
models += get_models_from_cond(negative, "gligen")
control_nets = get_models_from_cond(positive, "control") + get_models_from_cond(negative, "control")
gligen = get_models_from_cond(positive, "gligen") + get_models_from_cond(negative, "gligen")
gligen = [x[1] for x in gligen]
models = control_nets + gligen
comfy.model_management.load_controlnet_gpu(models)
return models

Loading…
Cancel
Save