Views
No views yet
[!WARNING] This model has only ~110,138,880 parameters and is trained on a small corpus. It is not guaranteed to perform well in production environments. Expect frequent hallucinations and limited factual knowledge.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("your_username/Nexara-Nano-100M-Chat", trust_remote_code=True)
4model = AutoModelForCausalLM.from_pretrained("your_username/Nexara-Nano-100M-Chat", trust_remote_code=True)
5
6# Format SFT prompt template
7prompt = "### System:\nYou are Nexara, a helpful and polite AI assistant.\n\n### User:\nWhat is your name?\n\n### Assistant:\n"
8inputs = tokenizer(prompt, return_tensors="pt")
9outputs = model.generate(**inputs, max_new_tokens=64, temperature=0.1)
10print(tokenizer.decode(outputs[0], skip_special_tokens=True))