Re:Accelerating Generative AI with PyTorch II: GPT, Fast
by dnnssl2 on Thursday November 30, 2023 @03:01PM
(#38478331)
Is this still the case for sliding window attention/streaming LLMs, where you have a fixed length attention window rather than infinitely passing in new tokens for quadratic scaling? You even get better performance due to purposely downsampling non-meaningful attention sink tokens.
[ Parent
| on HN ]
Re:Accelerating Generative AI with PyTorch II: GPT, Fast
by chillee on Thursday November 30, 2023 @03:12PM
(#38478450)
I cover it a bit in the blog post, but unless you have a really long context length (like 32k+), your primary computational cost doesn't come from attention but rather from loading your weights from VRAM into registers.
I mean, practically speaking, completions from say, ChatGPT or Claude take seconds to finish :)