nehcgs commited on
Commit
c7975a5
1 Parent(s): 74f2e45

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -192,7 +192,7 @@ import json
192
  from typing import Any, Dict, List
193
  from transformers import AutoModelForCausalLM, AutoTokenizer
194
 
195
- model_name = "katanemolabs/Arch-Function-1.5B"
196
  model = AutoModelForCausalLM.from_pretrained(
197
  model_name, device_map="auto", torch_dtype="auto", trust_remote_code=True
198
  )
 
192
  from typing import Any, Dict, List
193
  from transformers import AutoModelForCausalLM, AutoTokenizer
194
 
195
+ model_name = "katanemo/Arch-Function-1.5B"
196
  model = AutoModelForCausalLM.from_pretrained(
197
  model_name, device_map="auto", torch_dtype="auto", trust_remote_code=True
198
  )