[Fix] Try to fix error caused by latest cutedsl packages (#25690)

Co-authored-by: hnyls2002 <lsyincs@gmail.com>
This commit is contained in:
Baizhou Zhang
2026-05-18 16:51:32 -07:00
committed by GitHub
co-authored by hnyls2002
parent 314dedf7c6
commit b79e4b1e68
2 changed files with 21 additions and 4 deletions
+2 -2
View File
@@ -27,7 +27,7 @@ dependencies = [
"datasets",
"einops",
"fastapi",
"flashinfer_python==0.6.11.post1", # keep it aligned with jit-cache version in Dockerfile
"flashinfer_python[cu13]==0.6.11.post1", # keep it aligned with jit-cache version in Dockerfile
"flashinfer_cubin==0.6.11.post1",
"gguf",
"interegular",
@@ -37,7 +37,7 @@ dependencies = [
"ninja",
"easydict", # Required by remote model code (e.g. DeepSeek-OCR) loaded via trust_remote_code; validated by transformers 5.4+ check_imports
"numpy",
"nvidia-cutlass-dsl==4.5.0",
"nvidia-cutlass-dsl[cu13]==4.5.0",
"nvidia-ml-py",
"openai-harmony==0.0.4",
"openai==2.6.1",