Fix slow tests & sdpa dropout (#974)

Commit

3 years ago

Fix slow tests & sdpa dropout (#974) * fix slow tests & sdpa dropout * add comment