+onnxscript

mryzhov · mryzhov · commit 88ebd12c6052 · 2026-01-08T14:16:55.000+01:00
diff --git a/modules/token_merging/setup.py b/modules/token_merging/setup.py
@@ -4,7 +4,7 @@
 from setuptools import find_packages, setup
 
 EXTRAS_REQUIRE = {
-    "tests": ["onnx", "onnxruntime", "accelerate", "diffusers", "openvino", "optimum", "optimum-intel", "open-clip-torch","timm", "pytest"],
+    "tests": ["onnx", "onnxruntime", "onnxscript", "accelerate", "diffusers", "openvino", "optimum", "optimum-intel", "open-clip-torch","timm", "pytest"],
 }
 
 setup(

Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@`
`4`	`4`	`from setuptools import find_packages, setup`
`5`	`5`
`6`	`6`	`EXTRAS_REQUIRE = {`
`7`		`- "tests": ["onnx", "onnxruntime", "accelerate", "diffusers", "openvino", "optimum", "optimum-intel", "open-clip-torch","timm", "pytest"],`
	`7`	`+ "tests": ["onnx", "onnxruntime", "onnxscript", "accelerate", "diffusers", "openvino", "optimum", "optimum-intel", "open-clip-torch","timm", "pytest"],`
`8`	`8`	`}`
`9`	`9`
`10`	`10`	`setup(`