Titus 5ce90f3212 Bnb test fix for different hardwares (#29066) ... * generated text on A10G * generated text in CI * Apply suggestions from code review add explanatory comments Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> --------- Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>		2024-02-19 18:04:44 +00:00
..
benchmark
bettertransformer
deepspeed	fix failing trainer ds tests (#29057)	2024-02-16 17:18:45 +05:30
extended
fixtures
fsdp	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00
generation	ENH: added new output_logits option to generate function (#28667)	2024-02-19 17:34:17 +00:00
models	Fix the bert-base-cased tokenizer configuration test (#29105)	2024-02-19 13:23:25 +01:00
optimization
peft_integration
pipelines	Add chat support to text generation pipeline (#28945)	2024-02-16 16:41:01 +00:00
quantization	Bnb test fix for different hardwares (#29066)	2024-02-19 18:04:44 +00:00
repo_utils	Allow # Ignore copy (#27328)	2023-12-07 10:00:08 +01:00
sagemaker	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00
tokenization	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00
tools
trainer	Fix trainer test wrt DeepSpeed + auto_find_bs (#29061)	2024-02-16 10:04:24 -05:00
utils	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00
__init__.py
test_backbone_common.py	Align backbone stage selection with out_indices & out_features (#27606)	2023-12-20 18:33:17 +00:00
test_cache_utils.py	Fix static generation when compiling! (#28937)	2024-02-15 06:27:40 +01:00
test_configuration_common.py	[ PretrainedConfig] Improve messaging (#27438)	2023-11-15 14:10:39 +01:00
test_configuration_utils.py	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00
test_feature_extraction_common.py
test_feature_extraction_utils.py
test_image_processing_common.py	Fix a couple of typos and add an illustrative test (#26941)	2023-12-11 15:51:51 +00:00
test_image_processing_utils.py
test_image_transforms.py
test_modeling_common.py	Add tie_weights() to LM heads and set bias in set_output_embeddings() (#28948)	2024-02-14 20:39:01 +00:00
test_modeling_flax_common.py	[Flax] Update no init test for Flax v0.7.1 (#28735)	2024-01-26 18:20:39 +00:00
test_modeling_flax_utils.py	Enable safetensors conversion from PyTorch to other frameworks without the torch requirement (#27599)	2024-01-23 10:28:23 +01:00
test_modeling_tf_common.py	Add tf_keras imports to prepare for Keras 3 (#28588)	2024-01-30 17:26:36 +00:00
test_modeling_tf_utils.py	Add tf_keras imports to prepare for Keras 3 (#28588)	2024-01-30 17:26:36 +00:00
test_modeling_utils.py	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00
test_pipeline_mixin.py	Image Feature Extraction pipeline (#28216)	2024-02-05 14:50:07 +00:00
test_processing_common.py	Don't save processor_config.json if a processor has no extra attribute (#28584)	2024-01-19 09:59:14 +00:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00
test_tokenization_utils.py	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00