prithivMLmods
/

GWQ-9B-Preview

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

prithivMLmods commited on 28 days ago

Commit

a0f4c83

·

verified ·

1 Parent(s): 4de2400

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -21,7 +21,7 @@ GWQ - Gemma with Questions Prev is a family of lightweight, state-of-the-art ope
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b-it")
 model = AutoModelForCausalLM.from_pretrained(
     "prithivMLmods/GWQ-9B-Preview",
     device_map="auto",

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