PR #33918 BatchFeature.to() supports non-tensor keys

BatchFeature.to() supports non-tensor keys #33918

Rocketknight1 merged 9 commits into main from fix_oneformer_pipeline

Rocketknight1 added run-slow

Fix issue in oneformer preprocessing

7065c5e8

[run slow] oneformer

9ddbdfad

[run_slow] oneformer

38660964

Make the same fixes in DQA and object detection pipelines

502972f6

Fix BatchFeature.to() instead

2547a908

Revert pipeline-specific changes

4d796f72

Rocketknight1 force pushed to 4d796f72 1 year ago

Rocketknight1 changed the title ~~Fix issue in oneformer pipeline preprocessing~~ BatchFeature.to() supports non-tensor keys 1 year ago

LysandreJik approved these changes on 2024-10-07

Add the same check in Pixtral's methods

01fe6a83

Add the same check in BatchEncoding

64956bf6

make sure torch is imported

8e7ff3a6

LysandreJik approved these changes on 2024-10-08

Rocketknight1 merged fb360a6c into main 1 year ago

Rocketknight1 deleted the fix_oneformer_pipeline branch 1 year ago

Reviewers

LysandreJik

Assignees

No one assigned

Labels

run-slow

Milestone

No milestone