stockmark
/

stockmark-100b-instruct-v0.1

Inference Endpoints

Model card Files Files and versions Community

omitakahiro commited on May 15, 2024

Commit

57e0bba

•

1 Parent(s): f111a7f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -22,8 +22,8 @@ prompt_template = """### 指示:
 ### 応答:
 """
-tokenizer = AutoTokenizer.from_pretrained("stockmark/stockmark-100b")
-model = AutoModelForCausalLM.from_pretrained("stockmark/stockmark-100b", device_map="auto", torch_dtype=torch.bfloat16)
 instruction = "生成AIとは？"
 prompt = prompt_template.format(instruction=instruction)

 ### 応答:
 """
+tokenizer = AutoTokenizer.from_pretrained("stockmark/stockmark-100b-instruct-v0.1")
+model = AutoModelForCausalLM.from_pretrained("stockmark/stockmark-100b-instruct-v0.1", device_map="auto", torch_dtype=torch.bfloat16)
 instruction = "生成AIとは？"
 prompt = prompt_template.format(instruction=instruction)