transformers
fix(models): Fix ViVit when dealing with multiple GPUs
#43409
Merged

fix(models): Fix ViVit when dealing with multiple GPUs #43409

vasqu merged 1 commit into huggingface:main from tarekziade-fix-test_model_parallelism
tarekziade
tarekziade tarekziade requested a review from ydshieh ydshieh 237 days ago
HuggingFaceDocBuilderDev
tarekziade
vasqu
vasqu commented on 2026-01-23
tarekziade
tarekziade fix(models): Fix Vivit when dealing with multiple GPUs
d3fe46e0
tarekziade tarekziade changed the title fix(models): Fix ViVit and Vit modeling when dealing with multiple GPUs fix(models): Fix ViVit when dealing with multiple GPUs 236 days ago
github-actions
tarekziade
tarekziade tarekziade requested a review from vasqu vasqu 236 days ago
github-actions
github-actions
vasqu
vasqu approved these changes on 2026-01-23
vasqu vasqu merged 76ee621a into main 236 days ago
tarekziade tarekziade deleted the tarekziade-fix-test_model_parallelism branch 236 days ago
ydshieh

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone