Cyril Vallez f48ecd7608 Fix TP initialization (#35860) ...

* fix tp

* Update modeling_utils.py

* style

* style

* Update test_tp.py

* Update test_tp.py

* style

* Update test_tp.py

* Update test_tp.py

* Update test_tp.py

* Update test_tp.py

2025-01-28 15:07:37 +01: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

Qwen-2-5-VL: fix CI (#35935)

2025-01-28 14:51:57 +01:00

models

Fix mask slicing for models with HybridCache (#35681)

2025-01-28 14:35:00 +01:00

optimization

fix: Fixed the 1st argument name in classmethods (#31907)

2024-07-11 12:11:50 +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

Fix test_pipelines_video_classification that was always failing (#35842)

2025-01-23 19:22:32 +01: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

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

2025-01-28 11:38:45 +01: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