Update README.md
Browse files
README.md
CHANGED
|
@@ -44,6 +44,9 @@ Install the required libraries as follows:
|
|
| 44 |
|
| 45 |
Execute the following python code:
|
| 46 |
```python
|
|
|
|
|
|
|
|
|
|
| 47 |
tokenizer = AutoTokenizer.from_pretrained("pfnet/Qwen2.5-1.5B-pfn-qfin", trust_remote_code=True)
|
| 48 |
model = AutoModelForCausalLM.from_pretrained("pfnet/Qwen2.5-1.5B-pfn-qfin", device_map="auto", trust_remote_code=True)
|
| 49 |
text = "日本銀行は"
|
|
|
|
| 44 |
|
| 45 |
Execute the following python code:
|
| 46 |
```python
|
| 47 |
+
import torch
|
| 48 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 49 |
+
|
| 50 |
tokenizer = AutoTokenizer.from_pretrained("pfnet/Qwen2.5-1.5B-pfn-qfin", trust_remote_code=True)
|
| 51 |
model = AutoModelForCausalLM.from_pretrained("pfnet/Qwen2.5-1.5B-pfn-qfin", device_map="auto", trust_remote_code=True)
|
| 52 |
text = "日本銀行は"
|