• Jacky Lee's avatar
    Enable multi-device for some models (#30207) · 30b45320
    Jacky Lee authored
    
    
    * feat: multidevice for resnet
    
    * feat: yes! resnet
    
    * fix: compare all elements in tuple
    
    * feat: support for regnet
    
    * feat: support for convnextv2
    
    * feat: support for bit
    
    * feat: support for cvt
    
    * feat: add support for focalnet
    
    * feat: support for yolos
    
    * feat: support for glpn
    
    * feat: support for imagegpt
    
    * feat: support for levit
    
    * feat: support for mgp_str
    
    * feat: support for mobilnet_v1
    
    * feat: support for mobilnet_v2
    
    * feat: support for mobilevit
    
    * feat: support for mobilevitv2
    
    * feat: support for poolformer
    
    * fix: copies
    
    * fix: code quality check
    
    * update: upstream changes from main
    
    * fix: consistency check
    
    * feat: support for sam
    
    * feat: support for switchformer
    
    * feat: support for swin
    
    * feat: support for swinv2
    
    * feat: support for timesformer
    
    * feat: suport for trocr
    
    * feat: support for upernet
    
    * fix: check copies
    
    * update: rerun CI
    
    * update: rerun again, maybe
    
    * update: one more rerun
    
    ---------
    Co-authored-by: default avatarJacky Lee <jackylee328@gmail.com>
    30b45320
test_modeling_common.py 193 KB