[Deps] Use cu13 extra for nvidia cutlass dsl (#25576)

This commit is contained in:
Mohammad Miadh Angkad
2026-05-21 10:31:27 +08:00
committed by GitHub
parent 34479c19bd
commit a449ee4822
+1 -1
View File
@@ -38,7 +38,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[cu13]==4.5.0",
"nvidia-cutlass-dsl[cu13]==4.5.1",
"nvidia-ml-py",
"openai-harmony==0.0.4",
"openai==2.6.1",