E6E831728 commited on
Commit
d2524d0
·
verified ·
1 Parent(s): 667a9a7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -68,11 +68,11 @@ tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
68
  model = AutoModelForCausalLM.from_pretrained(repo_id, trust_remote_code=True)
69
  model.eval()
70
 
71
- prompt = "Question: What is the capital of United Kingdom?\nAnswer:"
72
  input_ids = torch.tensor([tokenizer.encode(prompt)], dtype=torch.long)
73
 
74
  with torch.no_grad():
75
- output_ids = model.generate(input_ids, max_new_tokens=16, do_sample=False)
76
 
77
  print(tokenizer.decode(output_ids[0].tolist()))
78
  ```
 
68
  model = AutoModelForCausalLM.from_pretrained(repo_id, trust_remote_code=True)
69
  model.eval()
70
 
71
+ prompt = "Question: What is the capital of France?\nAnswer:"
72
  input_ids = torch.tensor([tokenizer.encode(prompt)], dtype=torch.long)
73
 
74
  with torch.no_grad():
75
+ output_ids = model.generate(input_ids, max_new_tokens=3, do_sample=False)
76
 
77
  print(tokenizer.decode(output_ids[0].tolist()))
78
  ```