Safetensors
llama
vananh0905 commited on
Commit
164acdb
1 Parent(s): ad21cb9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -43,8 +43,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
43
  tokenizer = AutoTokenizer.from_pretrained("Fsoft-AIC/XMAiNframe-instruct-7b")
44
  model = AutoModelForCausalLM.from_pretrained("Fsoft-AIC/XMAiNframe-instruct-7b")
45
  messages=[
46
- {'role':'system','content':"You are a helpful assistant"},
47
- {'role': 'user', 'content': 'What is the future of Mainframe?'}
48
  ]
49
  inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
50
 
 
43
  tokenizer = AutoTokenizer.from_pretrained("Fsoft-AIC/XMAiNframe-instruct-7b")
44
  model = AutoModelForCausalLM.from_pretrained("Fsoft-AIC/XMAiNframe-instruct-7b")
45
  messages=[
46
+ {'from':'system', 'value': "You are a helpful assistant"},
47
+ {'from': 'human', 'value': 'What is the future of Mainframe?'}
48
  ]
49
  inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
50