Views
No views yet
1from transformers import AutoTokenizer, Qwen3MoeForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("kshitijthakkar/loggenix-moe-0.1B-e2-lr5e5-b4-3060-early-stopped")
4model = Qwen3MoeForCausalLM.from_pretrained("kshitijthakkar/loggenix-moe-0.1B-e2-lr5e5-b4-3060-early-stopped")
5
6# Example usage
7input_text = "def factorial(n):"
8inputs = tokenizer(input_text, return_tensors="pt")
9outputs = model.generate(inputs.input_ids, max_new_tokens=50)
10print(tokenizer.decode(outputs[0], skip_special_tokens=True))