from transformers import AutoTokenizer, AutoModelForCausalLM

model_name = "Chuxia-sys/Lora_tuned_planner"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")

prompt = "Plan a 3-day trip to Kyoto..."
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(**inputs, max_new_tokens=512)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

Chuxia-sys/lora-tuned-planner

Uploaded model

Lora_tuned_planner

Usage

Chuxia-sys/
lora-tuned-planner