PR #36460 Fix fp16 ONNX export for RT-DETR and RT-DETRv2

Fix fp16 ONNX export for RT-DETR and RT-DETRv2 #36460

qubvel merged 10 commits into huggingface:main from qubvel:fix-onnx-fp16-rt-detr

Fix FP16 ONNX export

00e2ead2

Fix typo

0a3fc6ed

qubvel commented on 2025-02-27

Sync omdet-turbo

cbbb776e

Refactor encoder for better readability

7e85bd15

qubvel commented on 2025-02-27

qubvel marked this pull request as ready for review 304 days ago

Fix _no_split_modules

12bd68ee

xenova approved these changes on 2025-02-28

Fix int -> torch_int

f1d74413

Fix rt_detr

4aefd1a8

Apply to rt-detr-v2

11ce2361

Fixup

1b1a901e

Fix copies

1b5a7b9b

shethaadit approved these changes on 2025-02-28

qubvel requested a review from

ArthurZucker 300 days ago

qubvel added Vision

ArthurZucker approved these changes on 2025-03-20

qubvel merged 1ddb6493 into main 283 days ago

Reviewers

ArthurZucker

xenova

shethaadit

Assignees

No one assigned

Labels

Vision

Milestone

No milestone