Function list_unets

scripts/trt.py:319–325

(l)

def list_unets(l):
    model = modelmanager.available_models()
    for k, v in model.items():
        if v[0]["config"].lora:
            continue
        label = "{} ({})".format(k, v[0]["base_model"]) if v[0]["config"].lora else k
        l.append(TrtUnetOption(label, v))


script_callbacks.on_list_unets(list_unets)