bjoernp commited on
Commit
ab1dc9b
1 Parent(s): b1f430b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -133,7 +133,7 @@ import torch
133
  device="cuda"
134
 
135
  model = AutoModelForCausalLM.from_pretrained(
136
- "DiscoResearch/Llama3-DiscoLeo-Instruct-8B-v0.1",
137
  torch_dtype="auto",
138
  device_map="auto"
139
  )
 
133
  device="cuda"
134
 
135
  model = AutoModelForCausalLM.from_pretrained(
136
+ "DiscoResearch/Llama3-DiscoLeo-Instruct-8B-32k",
137
  torch_dtype="auto",
138
  device_map="auto"
139
  )