Only adding 'head' dim to `plain_attention` function docstring

wayveai / Driving-with-LLMs

PyTorch implementation for the paper "Driving with LLMs: Fusing Object-Level Vector Modality for Explainable Autonomous Driving"

Apache License 2.0

408 stars 38 forks source link

Only adding 'head' dim to `plain_attention` function docstring #22

Open Anner-deJong opened 5 months ago

Anner-deJong commented 5 months ago

Was just reading your "Driving with LLMs: Fusing Object-Level Vector Modality for Explainable Autonomous Driving" paper :D and skimming through the code base a bit