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): def load_additional_models(positive, negative):
"""loads additional models in positive and negative conditioning""" """loads additional models in positive and negative conditioning"""
models = [] control_nets = get_models_from_cond(positive, "control") + get_models_from_cond(negative, "control")
models += get_models_from_cond(positive, "control") gligen = get_models_from_cond(positive, "gligen") + get_models_from_cond(negative, "gligen")
models += get_models_from_cond(negative, "control") gligen = [x[1] for x in gligen]
models += get_models_from_cond(positive, "gligen") models = control_nets + gligen
models += get_models_from_cond(negative, "gligen")
comfy.model_management.load_controlnet_gpu(models) comfy.model_management.load_controlnet_gpu(models)
return models return models

Loading…
Cancel
Save