Spaces:
Runtime error
Runtime error
crypto-code
commited on
Commit
•
7270a13
1
Parent(s):
15991c1
Update llama/m2ugen.py
Browse files- 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,
|