Update llama/m2ugen.py

llama/m2ugen.py

@@ -604,7 +604,7 @@ class M2UGen(nn.Module):
     def generate_music(self, embeddings, audio_length_in_s, music_caption):
         gen_prefix = ''.join([f'[AUD{i}]' for i in range(len(self.audio_tokens))])
         gen_prefx_ids = self.tokenizer(gen_prefix, add_special_tokens=False, return_tensors="pt").input_ids.to("cuda:1")
-        gen_prefix_embs = self.llama.tok_embeddings(gen_prefx_ids)
+        # gen_prefix_embs = self.llama.tok_embeddings(gen_prefx_ids)
         if self.music_decoder == "audioldm2":
             gen_emb = self.output_projector(embeddings.float().to("cuda:1"), gen_prefix_embs).squeeze(dim=0) / 10
             prompt_embeds, generated_prompt_embeds = gen_emb[:, :128 * 1024], gen_emb[:, 128 * 1024:]