Zach Mueller 4d8259d245 Fix loading zero3 weights (#36455) ...

* Check if fixes

* Fix zero3 loading

* Quality

* Fix marc nit

* Add fast tests

* Migrate to integrations.deepspeed rather than modeling_utils

* Style

2025-03-03 15:05:58 +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

Fix loading zero3 weights (#36455)

2025-03-03 15:05:58 +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

Universal Speculative Decoding CandidateGenerator (#35029)

2025-02-26 16:14:02 +00:00

models

Add Got-OCR 2 Fast image processor and refactor slow one (#36185)

2025-03-01 00:56:00 -05:00

optimization

Support constant lr with cooldown (#35453)

2025-02-10 13:21:55 +01:00

peft_integration

[tests] enable bnb tests on xpu (#36233)

2025-02-24 11:30:15 +01:00

pipelines

fix audio classification pipeline fp16 test on cuda (#36359)

2025-02-25 15:01:25 +01:00

quantization

Fix Expected output for compressed-tensors tests (#36425)

2025-02-26 21:17:24 +01:00

repo_utils

[Modular] skip modular checks based on diff (#36130)

2025-02-13 12:53:21 +00:00

sagemaker

Trainer - deprecate tokenizer for processing_class (#32385)

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

tokenization

Fix PretrainedTokenizerFast check => Fix PretrainedTokenizerFast Save (#35835)

2025-02-13 12:00:33 +01:00

tp

TP initialization module-by-module (#35996)

2025-02-19 14:04:57 +01:00

trainer

fix: prevent model access error during Optuna hyperparameter tuning (#36395)

2025-02-26 17:06:48 +01:00

utils

[generate] torch.distributed-compatible DynamicCache (#36373)

2025-02-27 11:48:57 +00: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_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

Au revoir flaky test_fast_is_faster_than_slow (#36240)

2025-02-17 18:30:07 +01:00

test_image_transforms.py

Uses Collection in transformers.image_transforms.normalize (#36301)

2025-02-21 18:38:41 +01:00

test_modeling_common.py

Fix hub_retry (#36449)

2025-02-27 14:38:25 +01:00

test_modeling_flax_common.py

[tests] remove flax-pt equivalence and cross tests (#36283)

2025-02-19 15:13:27 +00:00

test_modeling_tf_common.py

[tests] remove flax-pt equivalence and cross tests (#36283)

2025-02-19 15:13:27 +00:00

test_pipeline_mixin.py

Add image text to text pipeline (#34170)

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

test_processing_common.py

Uniformize LlavaNextVideoProcessor kwargs (#35613)

2025-02-18 14:13:51 -05:00

test_sequence_feature_extraction_common.py

Fix typo (#25966)

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

test_tokenization_common.py

[tests] remove pt_tf equivalence tests (#36253)

2025-02-19 11:55:11 +00:00

test_training_args.py

CI: fix test-save-trainer (#36191)

2025-02-14 10:20:56 +01:00