Spaces:
Runtime error
Runtime error
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,
|