Joaquin Caballero 031ef8802c fix FSDP + torch.compile bug when saving pretrained model (#37725) ...

* args keep_torch_compile=False in _save and _wwrap_method

* Fix FSDP execution on evaluation  for torch_compile mode

* add test trainer FSDP + Torch Compile

* fix quality code

* make style

* Revert " make style"

This reverts commit 77e797f8829c50992cc21496be3d9a3e480e1c97.

* make style

2025-05-06 17:51:28 +02:00

..

bettertransformer

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

deepspeed

🚨 rm already deprecated pad_to_max_length arg (#37617)

2025-05-01 15:21:55 +02:00

extended

Add Optional to remaining types (#37808)

2025-04-28 14:20:45 +01:00

fixtures

Implementation of SuperPoint and AutoModelForKeypointDetection (#28966)

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

fsdp

Fix the fsdp config cannot work issue. (#37549)

2025-04-28 10:44:51 +02:00

generation

Add GraniteMoeHybrid support for 4.0 (#37658)

2025-05-06 06:47:43 +02:00

models

[speech2text] fix init of sinusoidal embeddings (#37931)

2025-05-06 14:49:00 +01:00

optimization

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

peft_integration

FIX: Faulty PEFT tests (#37757)

2025-04-28 15:10:46 +02:00

pipelines

Process inputs directly in apply_chat_template in image-text-to-text pipeline (#35616)

2025-04-23 13:31:33 -04:00

quantization

Support AOPerModuleConfig and include_embedding (#37802)

2025-04-30 20:16:29 +02:00

repo_utils

Simplify soft dependencies and update the dummy-creation process (#36827)

2025-04-11 11:08:36 +02:00

sagemaker

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

tensor_parallel

enable tp on CPU (#36299)

2025-03-31 10:55:47 +02:00

tokenization

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

trainer

fix FSDP + torch.compile bug when saving pretrained model (#37725)

2025-05-06 17:51:28 +02:00

utils

[tests] Smaller model in slow cache tests (#37922)

2025-05-06 11:15:25 +01:00

__init__.py

…

test_backbone_common.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_configuration_common.py

Update composition flag usage (#36263)

2025-04-09 11:48:49 +02:00

test_feature_extraction_common.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_image_processing_common.py

Bridgetower fast image processor (#37373)

2025-04-16 22:39:18 +02:00

test_image_transforms.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_modeling_common.py

enable internvl UTs on XPU (#37779)

2025-04-30 10:29:40 +02:00

test_modeling_flax_common.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_modeling_tf_common.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_pipeline_mixin.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_processing_common.py

Fix typos in strings and comments (#37784)

2025-04-25 13:47:25 +01:00

test_sequence_feature_extraction_common.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_tokenization_common.py

🚨 rm already deprecated pad_to_max_length arg (#37617)

2025-05-01 15:21:55 +02:00

test_training_args.py

Fix TrainingArguments.torch_empty_cache_steps post_init check (#36734)

2025-03-17 16:09:46 +01:00