[Dinov2] Enable device_map="auto" support #38487
SunMarc
merged
8 commits into
huggingface:main
from
aryanchauhan31:clean-device-map-support-dinov2
Fix: resolve import order and duplicate import (ruff I001, F811)
669f5268
Format: clean up Dinov2 test file with ruff formatter
2a0605e1
Add _no_split_modules = ['Dinov2Layer'] to enable device_map='auto'
2b67df12
Revert dinov2_with_registers _no_split_modules to original state
3346f366
SunMarc
approved these changes
on 2025-06-03
Remove redundant device_map test as suggested
1dfb0588
Remove unused import after deleting test
ef13720e
SunMarc
commented
on 2025-06-03
removed import torch and the redundant test function
a98f7be5
SunMarc
commented
on 2025-06-04
Update tests/models/dinov2/test_modeling_dinov2.py
a91ba874
Login to write a write a comment.
Login via GitHub