Upload app.py with huggingface_hub

app.py

@@ -3,6 +3,9 @@ import os, sys, subprocess
 os.environ.setdefault("CUDA_HOME", "/usr/local/cuda")
 os.environ.setdefault("CONDA_PREFIX", "/usr/local")
 os.environ["LIDRA_SKIP_INIT"] = "true"
+os.environ["ATTN_BACKEND"] = "sdpa"
+os.environ["SPARSE_ATTN_BACKEND"] = "sdpa"
+os.environ["SPARSE_BACKEND"] = "spconv"
 
 # MUST import spaces before torch
 import spaces
@@ -20,7 +23,8 @@ if os.environ.get("HF_TOKEN"):
 STUB_KAOLIN = Path("/home/user/app/kaolin_stub")
 STUB_PT3D = Path("/home/user/app/pytorch3d_stub")
 STUB_SPCONV = Path("/home/user/app/spconv_stub")
-for stub in [STUB_KAOLIN, STUB_PT3D, STUB_SPCONV]:
+STUB_FA = Path("/home/user/app/flash_attn_stub")
+for stub in [STUB_KAOLIN, STUB_PT3D, STUB_SPCONV, STUB_FA]:
     if stub.exists():
         sys.path.insert(0, str(stub))
         print(f"Stub added: {stub.name}")