Views
No views yet
| Direction | BLEU (250 sentences from test split) |
|---|---|
| English -> Hinglish | 4.94 |
| Hinglish -> English | 17.89 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4# 1. Load from your Hugging Face Repo
5model_id = "MihaiPopa-1/Qwen3-0.6B-English-Hinglish-Preview-LoRA"
6
7tokenizer = AutoTokenizer.from_pretrained(model_id)
8model = AutoModelForCausalLM.from_pretrained(
9 model_id,
10 torch_dtype=torch.float32, # Standard for CPU
11 device_map="cpu" # Forces CPU usage
12)
13
14# 2. Translate (replace ron_Latn with your language here)
15prompt = "<|im_start|>user\nTranslate English to Hinglish: Hello, how are you doing?<|im_end|>\n<|im_start|>assistant\n"
16inputs = tokenizer(prompt, return_tensors="pt").to("cpu")
17
18with torch.no_grad():
19 outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.1)
20
21print(tokenizer.decode(outputs[0], skip_special_tokens=True))1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4# 1. Load from your Hugging Face Repo
5model_id = "MihaiPopa-1/Qwen3-0.6B-English-Hinglish-Preview-LoRA"
6
7tokenizer = AutoTokenizer.from_pretrained(model_id)
8model = AutoModelForCausalLM.from_pretrained(
9 model_id,
10 torch_dtype=torch.float32, # Standard for CPU
11 device_map="cpu" # Forces CPU usage
12)
13
14# 2. Translate (replace ron_Latn with your language here)
15prompt = "<|im_start|>user\nTranslate Hinglish to English: Hello, tum kaise ho?<|im_end|>\n<|im_start|>assistant\n"
16inputs = tokenizer(prompt, return_tensors="pt").to("cpu")
17
18with torch.no_grad():
19 outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.1)
20
21print(tokenizer.decode(outputs[0], skip_special_tokens=True))