Views
No views yet
Qwen/Qwen2.5-0.5B for the LunarLander action prediction task.Action: 0 (do nothing).Qwen/Qwen2.5-0.5BAli2023kosemen/lunarlander_balancedconversations1{
2 "conversations": [
3 {
4 "from": "human",
5 "value": "State: [x=..., y=..., vx=..., vy=..., angle=..., angular_vel=..., left_leg=..., right_leg=...]. What action should the lander take?"
6 },
7 {
8 "from": "gpt",
9 "value": "Action: 2 (fire main engine)."
10 }
11 ]
12}1024True16unslothadamw_8bit1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_id = "alirizaercan/qwen2.5-0.5b-lunarlander-action-lora"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(model_id)
7
8prompt = "State: [x=0.0005, y=1.4126, vx=0.0492, vy=0.0739, angle=-0.0006, angular_vel=-0.0112, left_leg=0.0000, right_leg=0.0000]. What action should the lander take?"
9inputs = tokenizer(prompt, return_tensors="pt")
10outputs = model.generate(**inputs, max_new_tokens=32)
11print(tokenizer.decode(outputs[0], skip_special_tokens=True))alirizaercanAli2023kosemen