[Fix] Try to fix error caused by latest cutedsl packages (#25690)
Co-authored-by: hnyls2002 <lsyincs@gmail.com>
This commit is contained in:
co-authored by
hnyls2002
parent
314dedf7c6
commit
b79e4b1e68
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user