crypto-code commited on
Commit
7270a13
1 Parent(s): 15991c1

Update llama/m2ugen.py

Browse files
Files changed (1) hide show
  1. llama/m2ugen.py +1 -0
llama/m2ugen.py CHANGED
@@ -620,6 +620,7 @@ class M2UGen(nn.Module):
620
  return audio_outputs
621
  else:
622
  print("Generating Music...")
 
623
  gen_inputs = self.generation_processor(text=music_caption, return_tensors="pt").to("cuda:1")
624
  #gen_emb = self.generation_model.generate(**gen_inputs, guidance_scale=3.5, encoder_only=True)
625
  audio_outputs = self.generation_model.generate(**gen_inputs, guidance_scale=3.5,
 
620
  return audio_outputs
621
  else:
622
  print("Generating Music...")
623
+ print(music_caption)
624
  gen_inputs = self.generation_processor(text=music_caption, return_tensors="pt").to("cuda:1")
625
  #gen_emb = self.generation_model.generate(**gen_inputs, guidance_scale=3.5, encoder_only=True)
626
  audio_outputs = self.generation_model.generate(**gen_inputs, guidance_scale=3.5,