maomaocun
/

dLLM-Var-no-template

Model card Files Files and versions

fix typo

#1

by zichenwen - opened Oct 11, 2025

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ for chunk in model.generate(
     block_length=64,
     threshold=0.9,
     streaming=True,
-    eos_token_id=tokenizer.eos_token,
 ):
     all_generated_ids = torch.cat([input_ids, chunk], dim=-1)
     text = tokenizer.batch_decode(all_generated_ids, skip_special_tokens=False)[0].split(tokenizer.eos_token)[0]

     block_length=64,
     threshold=0.9,
     streaming=True,
+    eos_token_id=tokenizer.eos_token_id,
 ):
     all_generated_ids = torch.cat([input_ids, chunk], dim=-1)
     text = tokenizer.batch_decode(all_generated_ids, skip_special_tokens=False)[0].split(tokenizer.eos_token)[0]