• 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