• Lei Wang's avatar
    [CI] Remove Flash Attention dependency (#705) · 87aae294
    Lei Wang authored
    * Update flash-attn version in requirements-test.txt from <=2.2.0 to ==2.5.8
    
    * lint fix
    
    * Remove unused dependencies from requirements-test.txt
    
    * Update import path for padding functions in example MHA forward variable length script
    
    * Refactor code formatting in bert_padding.py for improved readability
    87aae294
bert_padding.py 8.99 KB