--- library_name: transformers tags: [] --- # MISHANM/Indic_Multilingual_Llama-3-8B-Instruct This model is fine-tuned for the Indic languages , capable of answering queries and translating text from English to Indic languages . It leverages advanced natural language processing techniques to provide accurate and context-aware responses. ## Model Details This model is based on meta-llama/Llama-3.2-3B-Instruct and has been LoRA finetuned on Indic datasets: 1. Gujarati 2. Kannada 3. Hindi 4. Odia 5. Punjabi 6. Bengali 7. Tamil 8. Telugu # Training Details The model is trained on approx 246K instruction samples. 1. GPUs: 2*AMD Instinct MI210 2. train_samples_per_second': 33.055 3. train_steps_per_second': 1.625 4. train_loss': 0.25262785999764226, 5. epoch': 10.0 ## Inference with HuggingFace ```python3 import torch from transformers import AutoModelForCausalLM, AutoTokenizer # Set the device device = "cuda" if torch.cuda.is_available() else "cpu" # Load the fine-tuned model and tokenizer model_path = "MISHANM/Indic_Multilingual_Llama-3-8B-Instruct" model = AutoModelForCausalLM.from_pretrained(model_path) # Wrap the model with DataParallel if multiple GPUs are available if torch.cuda.device_count() > 1: print(f"Using {torch.cuda.device_count()} GPUs") model = torch.nn.DataParallel(model) # Move the model to the appropriate device model.to(device) tokenizer = AutoTokenizer.from_pretrained(model_path) # Function to generate text def generate_text(prompt, max_length=1000, temperature=0.9): # Format the prompt according to the chat template messages = [ { "role": "system", "content": "You are a language expert and linguist, with same knowledge give response in ().", #In place of "()" write your desired language in which response is required. }, {"role": "user", "content": prompt} ] # Apply the chat template formatted_prompt = f"<|system|>{messages[0]['content']}<|user|>{messages[1]['content']}<|assistant|>" # Tokenize and generate output inputs = tokenizer(formatted_prompt, return_tensors="pt").to(device) output = model.module.generate( # Use model.module for DataParallel **inputs, max_new_tokens=max_length, temperature=temperature, do_sample=True ) return tokenizer.decode(output[0], skip_special_tokens=True) # Example usage prompt = """ OME children were at play in their play-ground one day, when a herald rode through the town, blowing a trumpet, and crying aloud,.""" translated_text = generate_text(prompt) print(translated_text) ``` ## Citation Information ``` @misc{MISHANM/Indic_Multilingual_Llama-3-8B-Instruct, author = {Mishan Maurya}, title = {Introducing Fine Tuned LLM for Indic Languages}, year = {2024}, publisher = {Hugging Face}, journal = {Hugging Face repository}, } ```