transformers
Fix YOLOS device mismatch with device_map="auto"
#46886
Merged

Fix YOLOS device mismatch with device_map="auto" #46886

swankystark
swankystark Fix YOLOS device mismatch with device_map="auto"
b2e16509
qgallouedec Merge branch 'main' into fix/yolos-device-map-auto
945fdac7
molbap
molbap approved these changes on 2026-06-26
molbap
molbap requested changes on 2026-06-26
molbap molbap added Code agent slop
swankystark Set _no_split_modules for YOLOS and revert .to() workaround
7354f12b
swankystark
swankystark swankystark requested a review from molbap molbap 105 days ago
molbap
molbap approved these changes on 2026-08-31
molbap Merge branch 'main' into fix/yolos-device-map-auto
775a4dfe
molbap
github-actions
github-actions
molbap molbap enabled auto-merge 52 days ago
molbap Merge branch 'main' into fix/yolos-device-map-auto
e129e0c0
github-actions
github-actions
HuggingFaceDocBuilderDev
molbap molbap merged d9fe823d into main 43 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone