masanorihirano
commited on
Update README.md
Browse files
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/
|
46 |
-
model = AutoModelForCausalLM.from_pretrained("pfnet/
|
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():
|