katanemo
/

Arch-Function-3B

Text Generation

text-generation-inference

Model card Files Files and versions

nehcgs commited on Oct 8, 2024

Commit

ac4dd5d

·

verified ·

1 Parent(s): 78733a5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -192,7 +192,7 @@ import json
 from typing import Any, Dict, List
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_name = "katanemolabs/Arch-Function-3B"
 model = AutoModelForCausalLM.from_pretrained(
     model_name, device_map="auto", torch_dtype="auto", trust_remote_code=True
 )

 from typing import Any, Dict, List
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_name = "katanemo/Arch-Function-3B"
 model = AutoModelForCausalLM.from_pretrained(
     model_name, device_map="auto", torch_dtype="auto", trust_remote_code=True
 )