[BugFix] Fix tensorizer extra in setup.py (#4072)
This commit is contained in:
parent
2cd6b4f362
commit
563c54f760
2
setup.py
2
setup.py
@ -406,7 +406,7 @@ setup(
|
||||
install_requires=get_requirements(),
|
||||
ext_modules=ext_modules,
|
||||
extras_require={
|
||||
"optional": ["tensorizer==2.9.0a1"],
|
||||
"tensorizer": ["tensorizer==2.9.0a1"],
|
||||
},
|
||||
cmdclass={"build_ext": cmake_build_ext} if not _is_neuron() else {},
|
||||
package_data=package_data,
|
||||
|
Loading…
x
Reference in New Issue
Block a user