underspirit
commited on
Commit
•
3afe4cd
1
Parent(s):
9beb409
Update README.md
Browse files
README.md
CHANGED
@@ -111,6 +111,7 @@ C-Eval Category Results
|
|
111 |
The XVERSE-13B model can be loaded for inference using the following code:
|
112 |
|
113 |
```python
|
|
|
114 |
>>> from transformers import AutoTokenizer, AutoModelForCausalLM
|
115 |
>>> tokenizer = AutoTokenizer.from_pretrained("xverse/XVERSE-13B")
|
116 |
>>> model = AutoModelForCausalLM.from_pretrained("xverse/XVERSE-13B", trust_remote_code=True, torch_dtype=torch.float16, device_map='auto')
|
|
|
111 |
The XVERSE-13B model can be loaded for inference using the following code:
|
112 |
|
113 |
```python
|
114 |
+
>>> import torch
|
115 |
>>> from transformers import AutoTokenizer, AutoModelForCausalLM
|
116 |
>>> tokenizer = AutoTokenizer.from_pretrained("xverse/XVERSE-13B")
|
117 |
>>> model = AutoModelForCausalLM.from_pretrained("xverse/XVERSE-13B", trust_remote_code=True, torch_dtype=torch.float16, device_map='auto')
|