prithivMLmods
/

Blaze.1-27B-Preview

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

prithivMLmods commited on 29 days ago

Commit

be74ef7

·

verified ·

1 Parent(s): 84f0176

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ print(assistant_response)
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-27b-it")
 model = AutoModelForCausalLM.from_pretrained(
     "prithivMLmods/Blaze.1-27B-Preview",
     device_map="auto",

 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+tokenizer = AutoTokenizer.from_pretrained("prithivMLmods/Blaze.1-27B-Preview")
 model = AutoModelForCausalLM.from_pretrained(
     "prithivMLmods/Blaze.1-27B-Preview",
     device_map="auto",