Hemanth-thunder
/

Tamil-Mistral-7B-v0.1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Hemanth-thunder commited on Mar 16

Commit

35c0139

•

1 Parent(s): a877fa1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -37,7 +37,7 @@ Mistral-7B-v0.1 is a transformer model, with the following architecture choices:
 ```python
 import torch
 from transformers import (AutoModelForCausalLM,AutoTokenizer,TextStreamer,pipeline)
-model = AutoModelForCausalLM.from_pretrained("Hemanth-thunder/Tamil-Mistral-7B-v0.1"device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained("Hemanth-thunder/Tamil-Mistral-7B-v0.1",add_prefix_space=True)
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = "right"

 ```python
 import torch
 from transformers import (AutoModelForCausalLM,AutoTokenizer,TextStreamer,pipeline)
+model = AutoModelForCausalLM.from_pretrained("Hemanth-thunder/Tamil-Mistral-7B-v0.1",device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained("Hemanth-thunder/Tamil-Mistral-7B-v0.1",add_prefix_space=True)
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = "right"