"docs/source/vscode:/vscode.git/clone" did not exist on "f5c9738f61a4221e447bbf0c3a0496bd11583031"
Unverified Commit d95a32cc authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix torch version check in Vilt (#18260)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 7cb4da13
......@@ -41,7 +41,7 @@ from .configuration_vilt import ViltConfig
logger = logging.get_logger(__name__)
if torch.__version__ < (1, 10, 0):
if version.parse(torch.__version__) < version.parse("1.10.0"):
logger.warning(
f"You are using torch=={torch.__version__}, but torch>=1.10.0 is required to use "
"ViltModel. Please upgrade torch."
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment