DeepSeek/FlashMLA
1
0
Fork 0
mirror of https://github.com/deepseek-ai/FlashMLA synced 2025-06-26 18:15:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore(setup): properly package the repository as a Python package

Browse Source
This commit is contained in:
Xuehai Pan
2025-02-24 18:18:38 +08:00
parent 18e32770cc
commit 26d3077949
8 changed files with 108 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
flash_mla/flash_mla_interface.py
View File

@@ -2,7 +2,7 @@ from typing import Optional, Tuple
import torch
import flash_mla_cuda
from flash_mla import flash_mla_cuda
def get_mla_metadata(