thangvip commited on
Commit
b0c04d0
·
verified ·
1 Parent(s): 51a220d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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("nlplabtdtu/vilaw-sailor-1.8b-instruct", device_map="auto", cache_dir="./cache").eval()
44
- tokenizer = AutoTokenizer.from_pretrained("nlplabtdtu/vilaw-sailor-1.8b-instruct", cache_dir="./cache")
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"},