maywell commited on
Commit
aa97845
Β·
1 Parent(s): c268b2c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -58,8 +58,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
58
 
59
  device = "cuda" # the device to load the model onto
60
 
61
- model = AutoModelForCausalLM.from_pretrained("maywell/Synatra-7B-Instruct-v0.3")
62
- tokenizer = AutoTokenizer.from_pretrained("maywell/Synatra-7B-Instruct-v0.3")
63
 
64
  messages = [
65
  {"role": "user", "content": "λ°”λ‚˜λ‚˜λŠ” μ›λž˜ ν•˜μ–€μƒ‰μ΄μ•Ό?"},
 
58
 
59
  device = "cuda" # the device to load the model onto
60
 
61
+ model = AutoModelForCausalLM.from_pretrained("maywell/Synatra-7B-v0.3-base")
62
+ tokenizer = AutoTokenizer.from_pretrained("maywell/Synatra-7B-v0.3-base")
63
 
64
  messages = [
65
  {"role": "user", "content": "λ°”λ‚˜λ‚˜λŠ” μ›λž˜ ν•˜μ–€μƒ‰μ΄μ•Ό?"},