my-sd/modules_forge/forge_util.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

18 lines
424 B
Python
Raw Normal View History

2024-01-26 04:22:04 +00:00
from ldm_patched.modules.conds import CONDRegular, CONDCrossAttn
def cond_from_a1111_to_patched_ldm(cond):
cross_attn = cond['crossattn']
pooled_output = cond['vector']
result = dict(
cross_attn=cross_attn,
pooled_output=pooled_output,
model_conds=dict(
c_crossattn=CONDCrossAttn(cross_attn),
y=CONDRegular(pooled_output)
)
)
return [result, ]