From 606172f7b659b7f051981b488aa97bc6147994b8 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 20:26:20 -0800 Subject: [PATCH] Update forge_util.py --- modules_forge/forge_util.py | 10 ++++++++++ 1 file changed, 10 insertions(+) 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']