Fix slow tests & sdpa dropout (#974)

Commit

2 years ago

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