From 6015c4132f09c0606b58403d626bd0f42898c963 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Tue, 18 Jun 2024 21:24:10 -0700 Subject: [PATCH] Fix more unreferenced variables --- comfy/ldm/audio/autoencoder.py | 5 +++-- comfy/ldm/audio/dit.py | 6 +++++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/comfy/ldm/audio/autoencoder.py b/comfy/ldm/audio/autoencoder.py index e758b3ade..0daa6bd38 100644 --- a/comfy/ldm/audio/autoencoder.py +++ b/comfy/ldm/audio/autoencoder.py @@ -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 diff --git a/comfy/ldm/audio/dit.py b/comfy/ldm/audio/dit.py index cf8bb89b8..89af9a62c 100644 --- a/comfy/ldm/audio/dit.py +++ b/comfy/ldm/audio/dit.py @@ -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,