server/text_generation_server/layers/attention/cuda.py · 29b8d19cdf83d7fd26a4f17015228363a7163522 · OpenDAS / text-generation-inference · GitLab

Find file Blame History Permalink

fix: return the out tensor rather then the functions return value (#2361) · 29b8d19c
drbh authored Aug 06, 2024

29b8d19c

cuda.py 9.27 KB