Spaces:

tanyuzhou
/

Ro-Play

Running

tanyuzhou commited on Jun 11

Commit

2c7e5d9

•

1 Parent(s): 4cb7b09

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,8 +38,6 @@ def respond(
     inputs = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")
     inputs = inputs.to("cuda")
-    print("eos_token_id: ", tokenizer.eos_token_id)
     # Generate response
     generate_ids = model.generate(
         inputs,

     inputs = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")
     inputs = inputs.to("cuda")
     # Generate response
     generate_ids = model.generate(
         inputs,