兄弟们,Meta前几天甩出Llama 3.1 405B开源模型,号称性能对标GPT-4。我连夜在A100上跑了下,说点干货。
实测下来,405B的推理速度确实比上一代快30%左右,长上下文(128K)处理也稳,不会像某些模型到一半就崩。但别高兴太早——它显存需求炸裂,单卡A100 80G只能跑4-bit量化版,全精度部署至少8卡起步。想玩的话,推荐用vLLM或TGI,搭配FlashAttention-2,延迟能降到3秒内。
另外,RAG场景下,它跟LangChain集成不错,但注意别用默认chunk size,调成512 token效果更好。有个坑:中文分词偶尔抽风,建议先用Jieba预处理。
最后,别信官方吹的“零样本完胜”,我测了数学推理(GSM8K),它跟GPT-4还有5%差距,但开源能这样已经算卷王了。想省钱的,直接去HuggingFace下4-bit GGUF,单卡RTX 4090也能跑。
老规矩,部署遇到OOM不要慌,先调low_cpu_mem_usage=True。评论区见。 |