Spaces:

louisbrulenaudet
/

Pearl-7B

Sleeping

louisbrulenaudet commited on Mar 21

Commit

a93292a

•

1 Parent(s): 6ed5e3d

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -260,7 +260,8 @@ def generate(
         "do_sample": False,
         "num_beams": 1,
         "repetition_penalty": repetition_penalty,
-        "eos_token_id": tokenizer.eos_token_id
     }
     t = Thread(
@@ -321,6 +322,7 @@ with gr.Blocks() as demo:
     gr.Markdown(
         value=DESCRIPTION
     )
     chat_interface.render()
 if __name__ == "__main__":

         "do_sample": False,
         "num_beams": 1,
         "repetition_penalty": repetition_penalty,
+        "eos_token_id": tokenizer.eos_token_id,
+        "pad_token_id": tokenizer.eos_token_id
     }
     t = Thread(
     gr.Markdown(
         value=DESCRIPTION
     )
+    gr.DuplicateButton()
     chat_interface.render()
 if __name__ == "__main__":