Unverified Commit aebfc16b authored by Phil Wang's avatar Phil Wang Committed by GitHub
Browse files

Merge pull request #75 from zhvng/minor_fixes 

small bug fixes
parents 0ce7c656 018a37c9

audiolm_pytorch/audiolm_pytorch.py

+2 −1
Original line number Diff line number Diff line
@@ -1016,6 +1016,7 @@ class SemanticTransformerWrapper(nn.Module): 
            logits = self.transformer.forward_with_cond_scale(

                ids = sample_semantic_ids,

                text_embeds = text_embeds,

                cond_scale = cond_scale,

                **kwargs

            )
@@ -1600,7 +1601,7 @@ class AudioLM(nn.Module): 


        if self.needs_text:

            if exists(text):

                text_embeds = self.semantic.embed_text(texts)

                text_embeds = self.semantic.embed_text(text)



        if exists(prime_wave):

            prime_wave = prime_wave.to(self.device)