dev-mode-orpheus

Paused

Tomtom84 commited on Apr 20

Commit

f4406f3

verified ·

1 Parent(s): 83532d0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ NEW_BLOCK      = 128257
 EOS_TOKEN      = 128258
 AUDIO_BASE     = 128266
 AUDIO_SPAN   = 4096 * 7                                # 28 672 Codes
-VALID_AUDIO  = torch.arange(AUDIO_BASE, AUDIO_BASE + AUDIO_SPAN)
 # 2) Logit‑Mask (NEW_BLOCK + Audio; EOS erst nach 1. Block) ----------
 class AudioMask(LogitsProcessor):

 EOS_TOKEN      = 128258
 AUDIO_BASE     = 128266
 AUDIO_SPAN   = 4096 * 7                                # 28 672 Codes
+AUDIO_IDS  = torch.arange(AUDIO_BASE, AUDIO_BASE + AUDIO_SPAN) # Renamed VALID_AUDIO to AUDIO_IDS
 # 2) Logit‑Mask (NEW_BLOCK + Audio; EOS erst nach 1. Block) ----------
 class AudioMask(LogitsProcessor):