masanorihirano commited on
Commit
a3d24c5
·
verified ·
1 Parent(s): 4430cbf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -42,8 +42,8 @@ Install the required libraries as follows:
42
 
43
  Execute the following python code:
44
  ```python
45
- tokenizer = AutoTokenizer.from_pretrained("pfnet/nakomata-14b-pfn-qfin", trust_remote_code=True)
46
- model = AutoModelForCausalLM.from_pretrained("pfnet/nakomata-14b-pfn-qfin", device_map="auto", trust_remote_code=True)
47
  text = "日本銀行は"
48
  input_ids = tokenizer(text, return_tensors="pt").input_ids
49
  with torch.no_grad():
 
42
 
43
  Execute the following python code:
44
  ```python
45
+ tokenizer = AutoTokenizer.from_pretrained("pfnet/nekomata-14b-pfn-qfin", trust_remote_code=True)
46
+ model = AutoModelForCausalLM.from_pretrained("pfnet/nekomata-14b-pfn-qfin", device_map="auto", trust_remote_code=True)
47
  text = "日本銀行は"
48
  input_ids = tokenizer(text, return_tensors="pt").input_ids
49
  with torch.no_grad():