transformers/templates
LSinev 02b176c4ce
Fix torch version comparisons (#18460)
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
2022-08-03 13:37:18 -04:00
..
adding_a_missing_tokenization_test Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
adding_a_new_example_script Add examples telemetry (#17552) 2022-06-07 11:57:52 -04:00
adding_a_new_model Fix torch version comparisons (#18460) 2022-08-03 13:37:18 -04:00