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 :)
I mean, practically speaking, completions from say, ChatGPT or Claude take seconds to finish :)