transformers/tests/models/xglm
Damiano Amatruda 4d2de5f63c
Fix XGLM loss computation (PyTorch and TensorFlow) (#35878)
* Fix XGLM loss computation (PyTorch and TensorFlow)

* Update expected output string in XGLM sample test

This updates the expected output string of test_xglm_sample for torch
2.0 to the correct one and removes the one for torch 1.13.1 + cu116
(transformers moved to torch 2.0 with PR #35358).

* Update expected output IDs in XGLM generation test
2025-02-18 15:37:48 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_xglm.py [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
test_modeling_tf_xglm.py Fix XGLM loss computation (PyTorch and TensorFlow) (#35878) 2025-02-18 15:37:48 +01:00
test_modeling_xglm.py Fix XGLM loss computation (PyTorch and TensorFlow) (#35878) 2025-02-18 15:37:48 +01:00
test_tokenization_xglm.py Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00