From 16130c754671e25c94cd6a16c422b0bbae6d9e4e Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Fri, 3 Mar 2023 18:58:22 -0500 Subject: [PATCH] Add support for new colour T2I adapter model. --- comfy/sd.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/comfy/sd.py b/comfy/sd.py index 50d81f77..19722113 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -612,8 +612,17 @@ class T2IAdapter: def load_t2i_adapter(ckpt_path, model=None): t2i_data = load_torch_file(ckpt_path) - cin = t2i_data['conv_in.weight'].shape[1] - model_ad = adapter.Adapter(cin=cin, channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False) + keys = t2i_data.keys() + if "style_embedding" in keys: + pass + # TODO + # model_ad = adapter.StyleAdapter(width=1024, context_dim=768, num_head=8, n_layes=3, num_token=8) + elif "body.0.in_conv.weight" in keys: + cin = t2i_data['body.0.in_conv.weight'].shape[1] + model_ad = adapter.Adapter_light(cin=cin, channels=[320, 640, 1280, 1280], nums_rb=4) + else: + cin = t2i_data['conv_in.weight'].shape[1] + model_ad = adapter.Adapter(cin=cin, channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False) model_ad.load_state_dict(t2i_data) return T2IAdapter(model_ad, cin // 64)