from transformers import AutoModelForCausalLM, AutoTokenizer model_id = "Nanthasit/sakthai-context-0.5b-tools"tokenizer = AutoTokenizer.from_pretrained(model_id)model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto") tools = [ { "type": "function", "function": { "name": "get_weather", "description": "Get current weather", "parameters": { "type": "object", "properties": {"location": {"type": "string"}}, "required": ["location"], }, } }] messages = [ {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": "What's the weather in Tokyo?"},] text = tokenizer.apply_chat_template(messages, tools=tools, tokenize=False, add_generation_prompt=True)inputs = tokenizer([text], return_tensors="pt").to(model.device)outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.01, top_p=0.9)response = tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)print(response)