DriveFusion-V0.2 extends the Qwen2.5-VL architecture with a modular "Driving Intelligence" layer.
1import torch
2from drivefusion import DriveFusionForConditionalGeneration, DriveFusionProcessor
3
4# Load Model
5model_id = "DriveFusion/DriveFusion-V0.2"
6model = DriveFusionForConditionalGeneration.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
7processor = DriveFusionProcessor.from_pretrained(model_id)
8
9# Define Input: Image + Prompt + Telemetry
10gps_context = [[40.7128, -74.0060], [40.7130, -74.0058]] # Lat/Lon history
11speed_context = [[30.5]] # Current speed in m/s
12
13message = [{
14 "role": "user",
15 "content": [
16 {"type": "image", "image": "highway_scene.jpg"},
17 {"type": "text", "text": f"{'-->'*34}Analyze the scene and predict the next trajectory based on our current speed."}
18 ]
19}]
20
21# Generate
22inputs = processor(text=message, images="highway_scene.jpg", gps=gps_context, speed=speed_context, return_tensors="pt").to("cuda")
23output = model.generate(**inputs, max_new_tokens=512)
24
25# Results
26print("Reasoning:", output["text"])
27print("Predicted Trajectory (20 pts):", output["trajectory"])
28print("Target Speeds:", output["target_speeds"])
1@misc{drivefusionqa2026,
2 title={DriveFusion-V0.2: A Vision Language Action Model for Autonomous Driving},
3 author={Samir, Omar and DriveFusion Team},
4 year={2026},
5 url={https://huggingface.co/DriveFusion/DriveFusion-V0.2}
6}