feat: tpu flash attention working

AshishKumar4 · Aug 1, 2024 · 3018eab · 3018eab
1 parent bc697fa
commit 3018eab
Show file tree

Hide file tree

Showing 3 changed files with 558 additions and 251 deletions.
diff --git a/.gitignore b/.gitignore
@@ -9,4 +9,6 @@ good models
 tensorboard
 wandb
 gcs_mount
-datacache
+datacache
+*.deb
+gcsfuse.yml