Sumit Vij bc9a6d8302 Fix device mismatch error in Whisper model during feature extraction (#35866) ...

* Fix device mismatch error in whisper feature extraction

* Set default device

* Address code review feedback

---------

Co-authored-by: eustlb <94853470+eustlb@users.noreply.github.com>

2025-02-04 12:23:08 +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

Add GOT-OCR 2.0 to Transformers (#34721)

2025-01-31 11:28:13 -05:00

models

Fix device mismatch error in Whisper model during feature extraction (#35866)

2025-02-04 12:23:08 +01: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

Update-tp test (#35844)

2025-02-03 09:37:02 +01:00

trainer

layernorm_decay_fix (#35927)

2025-02-04 11:01:49 +01:00

utils

Add support for nested images to LLava and VipLLava (#35558)

2025-01-30 16:49:20 -05: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

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

Less flaky for TimmBackboneModelTest::test_batching_equivalence (#35971)

2025-01-30 16:56:26 +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

apply_chat_template: consistent behaviour for return_assistant_tokens_mask=True return_tensors=True (#35582)

2025-02-04 10:27:52 +01:00