travis0103
/

mistral_7b_paper_review_lora

Generated from Trainer

Model card Files Files and versions Community

travis0103 commited on Apr 19

Commit

5594d5a

•

1 Parent(s): 4b10f63

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -53,7 +53,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 messages = [
     {"role": "user", "content": SYSTEM_PROMPT},
-    {"role": "assistant", "content": abstract_input}
 ]
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 encoded_input = tokenizer.apply_chat_template(messages, return_tensors="pt").to(device)

 messages = [
     {"role": "user", "content": SYSTEM_PROMPT},
+    {"role": "user", "content": abstract_input}
 ]
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 encoded_input = tokenizer.apply_chat_template(messages, return_tensors="pt").to(device)