Spaces:

simonzhang5429
/

test3

Sleeping

simonzhang5429 commited on May 12

Commit

8800776

•

1 Parent(s): 2b7811f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ def generate(text):
     if text:
        chat_data.append({"role": "system", "content": text})
     print(chat_data)
-    achat=tokenizer.apply_chat_template(chat_data)
     print(achat)
     outputs = model.generate(achat,
                          max_new_tokens=128,

     if text:
        chat_data.append({"role": "system", "content": text})
     print(chat_data)
+    achat=tokenizer.apply_chat_template(chat_data,return_tensors="pt")
     print(achat)
     outputs = model.generate(achat,
                          max_new_tokens=128,