transformers/tests
Joao Gante f6e2586a36
Jamba: update integration tests (#32250)
* try test updates

* a few more changes

* a few more changes

* a few more changes

* [run slow] jamba

* skip logits checks on older gpus

* [run slow] jamba

* oops

* [run slow] jamba

* Update tests/models/jamba/test_modeling_jamba.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update tests/models/jamba/test_modeling_jamba.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-08-22 11:46:10 +01:00
..
agents Agents use grammar (#31735) 2024-08-07 11:42:52 +02:00
benchmark
bettertransformer
deepspeed Revert PR 32299, flag users when Zero-3 was missed (#32851) 2024-08-16 12:35:41 -04:00
extended Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
fixtures
fsdp 🚨🚨🚨 Update min version of accelerate to 0.26.0 (#32627) 2024-08-20 11:42:36 +02:00
generation Generate: unify LogitsWarper and LogitsProcessor (#32626) 2024-08-16 11:20:41 +01:00
models Jamba: update integration tests (#32250) 2024-08-22 11:46:10 +01:00
optimization fix: Fixed the 1st argument name in classmethods (#31907) 2024-07-11 12:11:50 +01:00
peft_integration
pipelines fix tensors on different devices in WhisperGenerationMixin (#32316) 2024-08-13 11:29:57 +01:00
quantization Cache: use batch_size instead of max_batch_size (#32657) 2024-08-16 11:48:45 +01:00
repo_utils
sagemaker Fixed log messages that are resulting in TypeError due to too many arguments (#32017) 2024-07-17 10:56:44 +01:00
tokenization #32184 save total_vocab_size (#32240) 2024-08-05 09:22:48 +02:00
trainer Add support for GrokAdamW optimizer (#32521) 2024-08-13 13:20:28 +01:00
utils Reduce the error log when using core models that need their weights renamed, and provide a step forward (#32656) 2024-08-16 13:05:57 -04:00
__init__.py
test_backbone_common.py
test_configuration_common.py Refactor: Removed un-necessary object base class (#32230) 2024-07-26 10:33:02 +02:00
test_feature_extraction_common.py
test_image_processing_common.py Update kwargs validation for preprocess with decorator (#32024) 2024-08-06 11:33:05 +01:00
test_image_transforms.py
test_modeling_common.py [tests] make test_sdpa_can_compile_dynamic device-agnostic (#32519) 2024-08-19 12:46:59 +01:00
test_modeling_flax_common.py
test_modeling_tf_common.py
test_pipeline_mixin.py fix: Fixed raising TypeError instead of ValueError for invalid type (#32111) 2024-07-22 17:46:17 +01:00
test_processing_common.py Modify ProcessorTesterMixin for better generalization (#32637) 2024-08-13 11:48:53 -04:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py Fix conflicting key in init kwargs in PreTrainedTokenizerBase (#31233) 2024-08-01 14:32:13 +02:00