π¨π¨π¨ Fix forward of Dinov2ForImageClassification for models with registers #37836
add num_tokens_to_discard to the forward of Dinov2ForImageClassification
ef2c1674
Merge branch 'main' into bugfix/discard-registers-in-dinov2-with-regiβ¦
51097d0d
qubvel
commented
on 2025-04-29
redefine forward in modular file, remove change to modeling_dinov2 file
d96a9ce8
Merge branch 'main' into bugfix/discard-registers-in-dinov2-with-regiβ¦
c3621368
run make fixup
794771ea
qubvel
commented
on 2025-04-30
Merge branch 'main' into bugfix/discard-registers-in-dinov2-with-regiβ¦
6b228298
qubvel
approved these changes
on 2025-05-05
qubvel
changed the title Fix forward of Dinov2ForImageClassification for models with registers π¨π¨π¨ Fix forward of Dinov2ForImageClassification for models with registers 316 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub