ibivibiv
/

multimaster-7b-v6

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

ibivibiv commited on Apr 19

Commit

afc5d53

•

1 Parent(s): c75bb57

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -134,8 +134,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 torch.set_default_device("cuda")
-model = AutoModelForCausalLM.from_pretrained("ibivibiv/multimaster-7b", torch_dtype="auto", device_config='auto')
-tokenizer = AutoTokenizer.from_pretrained("ibivibiv/multimaster-7b")
 inputs = tokenizer("### Instruction: Who would when in an arm wrestling match between Abraham Lincoln and Chuck Norris?\nA. Abraham Lincoln \nB. Chuck Norris\n### Response:\n", return_tensors="pt", return_attention_mask=False)

 torch.set_default_device("cuda")
+model = AutoModelForCausalLM.from_pretrained("ibivibiv/multimaster-7b-v6", torch_dtype="auto", device_config='auto')
+tokenizer = AutoTokenizer.from_pretrained("ibivibiv/multimaster-7b-v6")
 inputs = tokenizer("### Instruction: Who would when in an arm wrestling match between Abraham Lincoln and Chuck Norris?\nA. Abraham Lincoln \nB. Chuck Norris\n### Response:\n", return_tensors="pt", return_attention_mask=False)