fix torchvision version issue (#4406)

lanluo-nvidia · web-flow · commit dfcbf0ef7650 · 2026-07-14T15:47:46.000-07:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -90,7 +90,7 @@ test = [
 test-ext = [
     "timm>=1.0.3",
     "transformers>=5.4.0",
-    "torchvision>=0.29.0.dev,<0.30.0",
+    "torchvision>=0.28.0.dev,<0.29.0",
     "flashinfer-python; python_version >'3.9' and python_version <'3.13'",
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -90,7 +90,7 @@ test = [`
`90`	`90`	`test-ext = [`
`91`	`91`	`"timm>=1.0.3",`
`92`	`92`	`"transformers>=5.4.0",`
`93`		`- "torchvision>=0.29.0.dev,<0.30.0",`
	`93`	`+ "torchvision>=0.28.0.dev,<0.29.0",`
`94`	`94`	`"flashinfer-python; python_version >'3.9' and python_version <'3.13'",`
`95`	`95`	`]`
`96`	`96`