vllm/model_executor/layers/rotary_embedding.py · eb825c1e7401a6d9ebb2b3d8d693df0069b80ccb · kecinstone / 2024pra-vllm

Antoni Baum authored Nov 03, 2023


Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
Co-authored-by: Viktor Ferenczi <viktor@ferenczi.eu>
Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>

9f669a9a

rotary_embedding.py 10.4 KB

Replace rotary_embedding.py