ManishThota
/

CustomModel

Text Generation

text-generation-inference

Model card Files Files and versions Community

ManishThota commited on Mar 18, 2024

Commit

b9bc4aa

•

1 Parent(s): b42a347

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -31,11 +31,11 @@ torch.set_default_device("cuda")
 #Create model
 model = AutoModelForCausalLM.from_pretrained(
-    "ManishThota/Sparrow",
     torch_dtype=torch.float16,
     device_map="auto",
     trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained("ManishThota/SparrowVQE", trust_remote_code=True)
 #function to generate the answer
 def predict(question, image_path):

 #Create model
 model = AutoModelForCausalLM.from_pretrained(
+    "ManishThota/CustomModel",
     torch_dtype=torch.float16,
     device_map="auto",
     trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("ManishThota/CustomModel", trust_remote_code=True)
 #function to generate the answer
 def predict(question, image_path):