Enable multi-device for some models #30207
amyeroberts
merged
35 commits into
huggingface:main
from
jla524:resnet_multidevice
feat: multidevice for resnet
9962db7f
feat: yes! resnet
9292c985
fix: compare all elements in tuple
26cbabc5
feat: support for regnet
3623ddbd
jla524
changed the title
feat: support for convnextv2
a28789c7
jla524
changed the title
feat: support for bit
7bc5ca98
feat: support for cvt
de0ee8d0
feat: add support for focalnet
ed4c1ccf
feat: support for yolos
ad7f0239
feat: support for glpn
da150abd
feat: support for imagegpt
dff49721
feat: support for levit
6cc5243a
feat: support for mgp_str
d6cf3eb4
feat: support for mobilnet_v1
e122cf46
feat: support for mobilnet_v2
bb376dab
feat: support for mobilevit
7300b014
feat: support for mobilevitv2
32e5a453
feat: support for poolformer
65c75d82
amyeroberts
approved these changes
on 2024-04-17
fix: copies
249e9ec5
fix: code quality check
b015b9c9
update: upstream changes from main
d0c095c7
fix: consistency check
d5b7e838
Merge branch 'upstream' into resnet_multidevice
c45fd56d
feat: support for sam
0b97fab3
feat: support for switchformer
59d17da1
feat: support for swin
142b7494
feat: support for swinv2
54047fe4
feat: support for timesformer
d823e4dc
feat: suport for trocr
7f730c81
feat: support for upernet
64959f60
fix: check copies
268484c6
update: rerun CI
90f52e6c
update: rerun again, maybe
b470e5f4
Merge branch 'resnet_multidevice' of github.com:jla524/transformers i…
71d00343
update: one more rerun
a223c8ad
Login to write a write a comment.
Login via GitHub