YuE-music-generator-demo-zero

Paused

KingNish commited on Jan 30

Commit

af06be7

verified ·

1 Parent(s): 4c3b432

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -232,7 +232,7 @@ def generate_music(
                     repetition_penalty=repetition_penalty,
                     eos_token_id=mmtokenizer.eoa,
                     pad_token_id=mmtokenizer.eoa,
-                    # logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
                     guidance_scale=guidance_scale,
                     use_cache=True,
                     top_k=50,

                     repetition_penalty=repetition_penalty,
                     eos_token_id=mmtokenizer.eoa,
                     pad_token_id=mmtokenizer.eoa,
+                    logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
                     guidance_scale=guidance_scale,
                     use_cache=True,
                     top_k=50,