E6E831728 commited on
Commit
cbf1b13
·
verified ·
1 Parent(s): b9820e0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -75,7 +75,7 @@ tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
75
  model = AutoModelForCausalLM.from_pretrained(repo_id, trust_remote_code=True)
76
  model.eval()
77
 
78
- prompt = "Question: What is the capital of United Kingdom?\nAnswer:"
79
  input_ids = torch.tensor([tokenizer.encode(prompt)], dtype=torch.long)
80
 
81
  with torch.no_grad():
 
75
  model = AutoModelForCausalLM.from_pretrained(repo_id, trust_remote_code=True)
76
  model.eval()
77
 
78
+ prompt = "Question: What is the capital of UK?\nAnswer:"
79
  input_ids = torch.tensor([tokenizer.encode(prompt)], dtype=torch.long)
80
 
81
  with torch.no_grad():