From b51ff8a176e7eeb13d431161a4840c6bb4ccdaf6 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 15 Jun 2022 09:01:20 -0600 Subject: [PATCH] whoops! --- codes/models/audio/music/transformer_diffusion12.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/transformer_diffusion12.py b/codes/models/audio/music/transformer_diffusion12.py index 0abcfacb..372b7fc3 100644 --- a/codes/models/audio/music/transformer_diffusion12.py +++ b/codes/models/audio/music/transformer_diffusion12.py @@ -168,7 +168,7 @@ class TransformerDiffusion(nn.Module): for p in self.parameters(): p.DO_NOT_TRAIN = True p.requires_grad = False - for m in [self.input_converter and self.code_converter]: + for m in [self.ar_input and self.ar_prior_intg]: for p in m.parameters(): del p.DO_NOT_TRAIN p.requires_grad = True