Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -221,7 +221,7 @@ def generate_music(
|
|
221 |
logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
|
222 |
guidance_scale=guidance_scale,
|
223 |
use_cache=True,
|
224 |
-
num_beams =
|
225 |
)
|
226 |
if output_seq[0][-1].item() != mmtokenizer.eoa:
|
227 |
tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
|
|
|
221 |
logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
|
222 |
guidance_scale=guidance_scale,
|
223 |
use_cache=True,
|
224 |
+
num_beams = 1
|
225 |
)
|
226 |
if output_seq[0][-1].item() != mmtokenizer.eoa:
|
227 |
tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
|