CreatorFlow / Kho Repo / Dao-AILab/flash-attention

Dao-AILab/flash-attention

FlashAttention

LLM & ChatbotPython🔴 Nâng cao★ 25k

📖 Mô tả

FlashAttention là kỹ thuật giúp model AI xử lý văn bản dài nhanh hơn, tốn ít bộ nhớ hơn nhiều, gần như mọi model AI hiện đại đều dùng nó bên trong. Đây là thư viện cho dân làm AI, không phải app cho người dùng cuối: cài vào để model chạy nhanh hơn. Người mới chỉ cần biết tên nó khi thấy trong yêu cầu của công cụ khác. Lưu ý: chỉ chạy trên card đồ họa NVIDIA, lúc cài có thể biên dịch khá lâu.

📝 Tóm tắt README

FlashAttention là kỹ thuật giúp model AI xử lý văn bản dài nhanh hơn, tốn ít bộ nhớ hơn nhiều, gần như mọi model AI hiện đại đều dùng nó bên trong. Đây là thư viện cho dân làm AI, không phải app cho người dùng cuối: cài vào để model chạy nhanh hơn. Người mới chỉ cần biết tên nó khi thấy trong yêu cầu của công cụ khác. Lưu ý: chỉ chạy trên card đồ họa NVIDIA, lúc cài có thể biên dịch khá lâu.

✨ Tính năng nổi bật

▶ Dùng thử

▶ Dùng thử:
  1. B1: Cài: pip install flash-attn
  2. B2: Code: from flash_attn import flash_attn_func
  3. B3: Gọi hàm trong code huấn luyện hoặc suy luận của bạn theo ví dụ trong README (cần GPU NVIDIA)
Mở GitHub ↗
git clone https://github.com/Dao-AILab/flash-attention.git

🔗 Repo tương tự

← Quay lại Kho Repo