"megatron/model/vision/esvit_swin_backbone.py" did not exist on "f00d0a3fc1fb59b05669ca35ddecb6d422403bfd"
  • fxmarty's avatar
    Fix SDPA tests (#28552) · 2c1eebc1
    fxmarty authored
    
    
    * skip bf16 test if not supported by device
    
    * fix
    
    * fix bis
    
    * use is_torch_bf16_available_on_device
    
    * use is_torch_fp16_available_on_device
    
    * fix & use public llama
    
    * use 1b model
    
    * fix flacky test
    
    ---------
    Co-authored-by: default avatarYour Name <you@example.com>
    2c1eebc1
test_modeling_llama.py 30 KB