belyakoff commited on
Commit
8137326
·
verified ·
1 Parent(s): 17c541d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -22,7 +22,7 @@ tokenizer = AutoTokenizer.from_pretrained(checkpoint)
22
  # for multiple GPUs install accelerate and do `model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map="auto")`
23
  model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
24
 
25
- messages = [{"role": "user", "content": "What is the capital of France."}]
26
  input_text=tokenizer.apply_chat_template(messages, tokenize=False)
27
  print(input_text)
28
  inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
 
22
  # for multiple GPUs install accelerate and do `model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map="auto")`
23
  model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
24
 
25
+ messages = [{"role": "user", "content": "Столица России?"}]
26
  input_text=tokenizer.apply_chat_template(messages, tokenize=False)
27
  print(input_text)
28
  inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)