from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("damerajee/openhathi-h2e-e2h")
model = AutoModelForCausalLM.from_pretrained("damerajee/openhathi-h2e-e2h")
Inference
For english to hindi(e2h)
inputs = tokenizer(["[INST]translate this from english to hindi: Be a free thinker and don't accept everything you hear as truth. Be critical and evaluate what you believe in. [/INST]<s> hindi output:"]*1, return_tensors = "pt")
outputs = model.generate(**inputs, max_new_tokens = 18, use_cache = True)
tokenizer.batch_decode(outputs)
For hindi to english(h2e)
inputs = tokenizer(["[INST]translate this from hindi to english: अगर तुम सूरज की तरह चमकना चाहते हो, तो सूरज की तरह जलना सीखो।[/INST]<s> english output:"]*1, return_tensors = "pt")
outputs = model.generate(**inputs, max_new_tokens = 18, use_cache = True)
tokenizer.batch_decode(outputs)
Dataset
The dataset used was the combination of two dataset which gave a total of 1_786_788 rows
The rows were then pre-process to look something like this :
[INST]translate this from english to hindi: When it is said to him: \'Fear Allah\' egotism takes him in his sin. Gehenna (Hell) shall be enough for him. How evil a cradling! [/INST] hindi output: और जब उससे कहा जाता है,"अल्लाह से डर", तो अहंकार उसे और गुनाह पर जमा देता है। अतः उसके लिए तो जहन्नम ही काफ़ी है, और वह बहुत-ही बुरी शय्या है! '
This was done for both english to hindi and hindi to english hence the name h2e and e2h
Now when doing the above we get a total of 3 million plus rows
Training details
The model was loaded in 4-Bit
The target modules include "q_proj", "k_proj", "v_proj", "o_proj"
The fine-tuning was done on a free goggle colab with a single t4 GPU (huge thanks to unsloth for this)
Even though the Full dataset was almost 3 million The lora model was finetuned on only 1 million row for each language
Limitations
The model was not fully trained on all the dataset and Much evaluation was not done so any contributions will be helpful.
As of right now this is a smaller model Better model trained on better dataset will be released
Runs of damerajee openhathi-h2e-e2h-small on huggingface.co
13
Total runs
0
24-hour runs
0
3-day runs
1
7-day runs
2
30-day runs
More Information About openhathi-h2e-e2h-small huggingface.co Model
openhathi-h2e-e2h-small huggingface.co is an AI model on huggingface.co that provides openhathi-h2e-e2h-small's model effect (), which can be used instantly with this damerajee openhathi-h2e-e2h-small model. huggingface.co supports a free trial of the openhathi-h2e-e2h-small model, and also provides paid use of the openhathi-h2e-e2h-small. Support call openhathi-h2e-e2h-small model through api, including Node.js, Python, http.
openhathi-h2e-e2h-small huggingface.co is an online trial and call api platform, which integrates openhathi-h2e-e2h-small's modeling effects, including api services, and provides a free online trial of openhathi-h2e-e2h-small, you can try openhathi-h2e-e2h-small online for free by clicking the link below.
damerajee openhathi-h2e-e2h-small online free url in huggingface.co:
openhathi-h2e-e2h-small is an open source model from GitHub that offers a free installation service, and any user can find openhathi-h2e-e2h-small on GitHub to install. At the same time, huggingface.co provides the effect of openhathi-h2e-e2h-small install, users can directly use openhathi-h2e-e2h-small installed effect in huggingface.co for debugging and trial. It also supports api for free installation.
openhathi-h2e-e2h-small install url in huggingface.co: