Hugging Face's logo

ManishThota
/
CustomModel

Text Generation
Transformers
PyTorch
Safetensors
English
Sparrow
endpoints
text-generation-inference
custom_code
Model card Files
xet
 Community
6
ManishThota commited on
Commit
b9bc4aa
·
verified ·
1 Parent(s): b42a347

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -31,11 +31,11 @@ torch.set_default_device("cuda")
31
 
32
  #Create model
33
  model = AutoModelForCausalLM.from_pretrained(
34
- "ManishThota/Sparrow",
35
  torch_dtype=torch.float16,
36
  device_map="auto",
37
  trust_remote_code=True)
38
- tokenizer = AutoTokenizer.from_pretrained("ManishThota/SparrowVQE", trust_remote_code=True)
39
 
40
  #function to generate the answer
41
  def predict(question, image_path):
 
31
 
32
  #Create model
33
  model = AutoModelForCausalLM.from_pretrained(
34
+ "ManishThota/CustomModel",
35
  torch_dtype=torch.float16,
36
  device_map="auto",
37
  trust_remote_code=True)
38
+ tokenizer = AutoTokenizer.from_pretrained("ManishThota/CustomModel", trust_remote_code=True)
39
 
40
  #function to generate the answer
41
  def predict(question, image_path):