Update forge_util.py

This commit is contained in:
lllyasviel 2024-01-25 20:26:20 -08:00
parent ed495d7286
commit 606172f7b6

View File

@ -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']