Spaces:

maxidl
/

openreviewer

Running on Zero

maxidl commited on Dec 11, 2024

Commit

78dca84

1 Parent(s): 771dd42

a

Files changed (1) hide show

app.py CHANGED Viewed

@@ -193,7 +193,7 @@ def generate(paper_text, review_template):
     generated_text = ""
     for new_text in streamer:
         generated_text += new_text
-        yield generated_text
     # generated_ids = model.generate(
     #     input_ids=input_ids,
     #     max_new_tokens=4096,

     generated_text = ""
     for new_text in streamer:
         generated_text += new_text
+        yield generated_text.replace("<|eot_id|>", "")
     # generated_ids = model.generate(
     #     input_ids=input_ids,
     #     max_new_tokens=4096,