Jonas Rohw 23d782ead2 Output dicts support in text generation pipeline (#35092) ...

* Support for generate_argument: return_dict_in_generate=True, instead of returning a error

* fix: call test with return_dict_in_generate=True

* fix: Only import torch if it is present

* update: Encapsulate output_dict changes

* fix: added back original comments

---------

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

2025-01-29 14:44:46 +00:00

..

agents

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

bettertransformer

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

deepspeed

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

extended

[tests] skip tests for xpu (#33553)

2024-09-19 19:28:04 +01:00

fixtures

Implementation of SuperPoint and AutoModelForKeypointDetection (#28966)

2024-03-19 14:43:02 +00:00

fsdp

[tests] make cuda-only tests device-agnostic (#35607)

2025-01-13 14:48:39 +01:00

generation

Fix flaky test_assisted_decoding_matches_greedy_search (#35951)

2025-01-29 14:50:07 +01:00

models

Test: generate with torch.compile(model.forward) as a fast test (#34544)

2025-01-28 14:10:38 +00:00

optimization

Update unwrap_and_save_reload_schedule to use weights_only=False (#35952)

2025-01-29 14:30:57 +01:00

peft_integration

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

pipelines

Output dicts support in text generation pipeline (#35092)

2025-01-29 14:44:46 +00:00

quantization

Split and clean up GGUF quantization tests (#35502)

2025-01-27 15:46:57 +01:00

repo_utils

Fix modular edge case + modular sorting order (#35562)

2025-01-09 17:17:52 +01:00

sagemaker

Trainer - deprecate tokenizer for processing_class (#32385)

2024-10-02 14:08:46 +01:00

tokenization

tokenizer train from iterator without pre_tokenizers (#35396)

2025-01-09 15:34:43 +01:00

tp

Fix TP initialization (#35860)

2025-01-28 15:07:37 +01:00

trainer

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

utils

Test: generate with torch.compile(model.forward) as a fast test (#34544)

2025-01-28 14:10:38 +00:00

__init__.py

GPU text generation: mMoved the encoded_prompt to correct device

2020-01-06 15:11:12 +01:00

test_backbone_common.py

Align backbone stage selection with out_indices & out_features (#27606)

2023-12-20 18:33:17 +00:00

test_configuration_common.py

Load sub-configs from composite configs (#34410)

2024-11-05 11:34:01 +01:00

test_feature_extraction_common.py

Split common test from core tests (#24284)

2023-06-15 07:30:24 -04:00

test_image_processing_common.py

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

test_image_transforms.py

fix: center_crop occasionally outputs off-by-one dimension matrix (#30934)

2024-05-21 13:56:52 +01:00

test_modeling_common.py

Fix: loading DBRX back from saved path (#35728)

2025-01-28 11:38:45 +01:00

test_modeling_flax_common.py

🚨All attention refactor🚨 (#35235)

2024-12-18 16:53:39 +01:00

test_modeling_tf_common.py

🚨All attention refactor🚨 (#35235)

2024-12-18 16:53:39 +01:00

test_pipeline_mixin.py

Add image text to text pipeline (#34170)

2024-10-31 15:48:11 -04:00

test_processing_common.py

VLMs: major clean up 🧼 (#34502)

2025-01-08 10:35:23 +01:00

test_sequence_feature_extraction_common.py

Fix typo (#25966)

2023-09-05 10:12:25 +02:00

test_tokenization_common.py

[tokenizers] Ensure that add_prefix_space is propagated to backend_tokenizer.pre_tokenizer (#35593)

2025-01-09 17:46:50 +01:00