wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-09 04:52:32 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Fix more unreferenced variables

Browse Source
This commit is contained in:
doctorpangloss 2024-06-18 21:24:10 -07:00
parent 2aecff06ff
commit 6015c4132f
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/ldm/audio/autoencoder.py
View File

@ -96,8 +96,9 @@ def get_activation(activation: Literal["elu", "snake", "none"], antialias=False,
else: else:
raise ValueError(f"Unknown activation {activation}") raise ValueError(f"Unknown activation {activation}")
if antialias: # todo: ??? Not existing
act = Activation1d(act) # if antialias:
# act = Activation1d(act)
return act return act

6
comfy/ldm/audio/dit.py
View File

@ -775,6 +775,8 @@ class AudioDiffusionTransformer(nn.Module):
return_info=False, return_info=False,
**kwargs): **kwargs):
info = None
output = None
if cross_attn_cond is not None: if cross_attn_cond is not None:
cross_attn_cond = self.to_cond_embed(cross_attn_cond) cross_attn_cond = self.to_cond_embed(cross_attn_cond)
@ -871,8 +873,10 @@ class AudioDiffusionTransformer(nn.Module):
mask=None, mask=None,
return_info=False, return_info=False,
control=None, control=None,
transformer_options={}, transformer_options=None,
**kwargs): **kwargs):
if transformer_options is None:
transformer_options = {}
return self._forward( return self._forward(
x, x,
timestep, timestep,