Text Generation
Transformers
Safetensors
Arabic
llama
arabic
reasoning
chain-of-thought
math
gsm8k
small-language-model
slm
sft
conversational
text-generation-inference
oddadmix's picture
training code: data generation, SFT, eval, GRPO
867d0f3 verified
Raw
History Blame Contribute Delete
1.01 kB
"""Quick manual probe: python chat.py "ุณุคุงู„ูƒ ู‡ู†ุง" [model_dir]"""
import sys, torch
from transformers import AutoModelForCausalLM, AutoTokenizer
question = sys.argv[1] if len(sys.argv) > 1 else "ุฅุฐุง ูƒุงู† ู„ุฏูŠูƒ 1500 ุฑูŠุงู„ ูˆุฃู†ูู‚ุช 20% ู…ู†ู‡ุง ุนู„ู‰ ุงู„ูƒุชุจุŒ ููƒู… ุชุจู‚ู‰ ู…ุนูƒุŸ"
model_dir = sys.argv[2] if len(sys.argv) > 2 else "./Nawah-Reasoning-v1"
tok = AutoTokenizer.from_pretrained(model_dir)
model = AutoModelForCausalLM.from_pretrained(model_dir, dtype=torch.bfloat16).cuda().eval()
prompt = tok.apply_chat_template([{"role": "user", "content": question}], tokenize=False, add_generation_prompt=True)
ids = tok(prompt, return_tensors="pt").to("cuda")
out = model.generate(**ids, max_new_tokens=512, do_sample=False,
eos_token_id=[tok.eos_token_id, tok.convert_tokens_to_ids("<|im_end|>")],
pad_token_id=tok.pad_token_id)
print(tok.decode(out[0][ids["input_ids"].shape[1]:], skip_special_tokens=False).split("<|im_end|>")[0])