diff --git a/modules_forge/forge_util.py b/modules_forge/forge_util.py index c8000890..9caa84a9 100644 --- a/modules_forge/forge_util.py +++ b/modules_forge/forge_util.py @@ -1,7 +1,17 @@ +import torch from ldm_patched.modules.conds import CONDRegular, CONDCrossAttn def cond_from_a1111_to_patched_ldm(cond): + if isinstance(cond, torch.Tensor): + result = dict( + cross_attn=cond, + model_conds=dict( + c_crossattn=CONDCrossAttn(cond), + ) + ) + return [result, ] + cross_attn = cond['crossattn'] pooled_output = cond['vector']