Spaces:

mrfakename
/

rwkv-music

Running on Zero

mrfakename commited on Feb 20

Commit

f9283c8

•

1 Parent(s): d3a6851

Update musiclib.py

Files changed (1) hide show

musiclib.py CHANGED Viewed

@@ -62,7 +62,7 @@ def musicgen(ccc='<pad>', piano_only=False):
             else:
                 occurrence[token] = 0.3 + (occurrence[token] if token in occurrence else 0)
-        output += TOKEN_SEP + tokenizer.decode([token]))
         yield output
     output += (' <end>')

             else:
                 occurrence[token] = 0.3 + (occurrence[token] if token in occurrence else 0)
+        output += TOKEN_SEP + tokenizer.decode([token])
         yield output
     output += (' <end>')