Update app.py
Browse files
app.py
CHANGED
@@ -220,16 +220,16 @@ def generate_music(
|
|
220 |
max_new_tokens=max_new_tokens,
|
221 |
min_new_tokens=100,
|
222 |
do_sample=False, # just testing what happens
|
223 |
-
top_p=top_p,
|
224 |
-
temperature=temperature,
|
225 |
-
repetition_penalty=repetition_penalty,
|
226 |
eos_token_id=mmtokenizer.eoa,
|
227 |
pad_token_id=mmtokenizer.eoa,
|
228 |
logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
|
229 |
-
guidance_scale=guidance_scale,
|
230 |
use_cache=True,
|
231 |
# top_k=50,
|
232 |
-
num_beams=1
|
233 |
)
|
234 |
if output_seq[0][-1].item() != mmtokenizer.eoa:
|
235 |
tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
|
|
|
220 |
max_new_tokens=max_new_tokens,
|
221 |
min_new_tokens=100,
|
222 |
do_sample=False, # just testing what happens
|
223 |
+
# top_p=top_p,
|
224 |
+
# temperature=temperature,
|
225 |
+
# repetition_penalty=repetition_penalty,
|
226 |
eos_token_id=mmtokenizer.eoa,
|
227 |
pad_token_id=mmtokenizer.eoa,
|
228 |
logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
|
229 |
+
# guidance_scale=guidance_scale,
|
230 |
use_cache=True,
|
231 |
# top_k=50,
|
232 |
+
# num_beams=1
|
233 |
)
|
234 |
if output_seq[0][-1].item() != mmtokenizer.eoa:
|
235 |
tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
|