Fix more unreferenced variables

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

View File

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

View File

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