transformers
a6937898 - Fix torch version comparisons

Commit
3 years ago
Fix torch version comparisons Comparisons like version.parse(torch.__version__) > version.parse("1.6") are True for torch==1.6.0+cu101 or torch==1.6.0+cpu version.parse(version.parse(torch.__version__).base_version) are preferred (and available in pytorch_utils.py
Author
Parents
Loading