JeongwonChoi commited on
Commit
0c5f306
โ€ข
1 Parent(s): 0e953f6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -76,8 +76,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
76
 
77
  device = "cuda" # the device to load the model onto
78
 
79
- model = AutoModelForCausalLM.from_pretrained("Edentns/DataVortexS-10.7B-dpo-v1.8")
80
- tokenizer = AutoTokenizer.from_pretrained("Edentns/DataVortexS-10.7B-dpo-v1.8")
81
 
82
  messages = [
83
  {"role": "system", "content": "๋‹น์‹ ์€ ์‚ฌ๋žŒ๋“ค์ด ์ •๋ณด๋ฅผ ์ฐพ์„ ์ˆ˜ ์žˆ๋„๋ก ๋„์™€์ฃผ๋Š” ์ธ๊ณต์ง€๋Šฅ ๋น„์„œ์ž…๋‹ˆ๋‹ค."},
 
76
 
77
  device = "cuda" # the device to load the model onto
78
 
79
+ model = AutoModelForCausalLM.from_pretrained("Edentns/DataVortexS-10.7B-dpo-v1.1")
80
+ tokenizer = AutoTokenizer.from_pretrained("Edentns/DataVortexS-10.7B-dpo-v1.1")
81
 
82
  messages = [
83
  {"role": "system", "content": "๋‹น์‹ ์€ ์‚ฌ๋žŒ๋“ค์ด ์ •๋ณด๋ฅผ ์ฐพ์„ ์ˆ˜ ์žˆ๋„๋ก ๋„์™€์ฃผ๋Š” ์ธ๊ณต์ง€๋Šฅ ๋น„์„œ์ž…๋‹ˆ๋‹ค."},