Update README.md
Browse files
README.md
CHANGED
|
@@ -40,8 +40,8 @@ Use the code below to get started with the model.
|
|
| 40 |
|
| 41 |
```python
|
| 42 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 43 |
-
model = AutoModelForCausalLM.from_pretrained("
|
| 44 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 45 |
|
| 46 |
messages = [
|
| 47 |
{'role': 'system', 'content': "bạn là trợ lý AI hữu ích"},
|
|
|
|
| 40 |
|
| 41 |
```python
|
| 42 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 43 |
+
model = AutoModelForCausalLM.from_pretrained("thangvip/vwen-1.5B-instruct", device_map="auto", cache_dir="./cache").eval()
|
| 44 |
+
tokenizer = AutoTokenizer.from_pretrained("thangvip/vwen-1.5B-instruct", cache_dir="./cache")
|
| 45 |
|
| 46 |
messages = [
|
| 47 |
{'role': 'system', 'content': "bạn là trợ lý AI hữu ích"},
|