Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2tok = AutoTokenizer.from_pretrained("woojin93/qwen0_5b-gsm8k-20251101-1022")
3m = AutoModelForCausalLM.from_pretrained("woojin93/qwen0_5b-gsm8k-20251101-1022", torch_dtype="auto", device_map="auto")
4
5prompt = "<|im_start|>user\nSolve: 12 + 35 = ?\n<|im_end|>\n<|im_start|>assistant\n"
6out = m.generate(**tok(prompt, return_tensors='pt').to(m.device), max_new_tokens=64)
7print(tok.decode(out[0], skip_special_tokens=True))