wkshin89
/

Yi-Ko-6B-Instruct-v1.1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

wkshin89 commited on Feb 18

Commit

cb396dd

•

1 Parent(s): 1909454

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -9,7 +9,7 @@ language:
 base_model: beomi/Yi-Ko-6B
 ---
-# Yi-Ko-6B-Instruct-v1.1_
 ## Model Details
@@ -36,9 +36,9 @@ base_model: beomi/Yi-Ko-6B
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained("wkshin89/Yi-Ko-6B-Instruct-v1.1_")
 model = AutoModelForCausalLM.from_pretrained(
-    "wkshin89/Yi-Ko-6B-Instruct-v1.1_",
     device_map="auto",
     torch_dtype=torch.bfloat16,
 )

 base_model: beomi/Yi-Ko-6B
 ---
+# Yi-Ko-6B-Instruct-v1.1
 ## Model Details
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("wkshin89/Yi-Ko-6B-Instruct-v1.1")
 model = AutoModelForCausalLM.from_pretrained(
+    "wkshin89/Yi-Ko-6B-Instruct-v1.1",
     device_map="auto",
     torch_dtype=torch.bfloat16,
 )