Upload folder using huggingface_hub

app.py

@@ -40,25 +40,23 @@ def apply_sage_attention_patch():
 @spaces.GPU
 def dummy_gpu_for_startup():
     print("--- [GPU Startup] Dummy function for startup check initiated. ---")
-    # SageAttention patch temporarily disabled for faster iteration.
-    # patch_result = apply_sage_attention_patch()
-    # print(f"--- [GPU Startup] {patch_result} ---")
+    patch_result = apply_sage_attention_patch()
+    print(f"--- [GPU Startup] {patch_result} ---")
     print("--- [GPU Startup] Startup check passed. ---")
     return "Startup check passed."
 
 def main():
     from utils.app_utils import print_welcome_message
-    # from scripts import build_sage_attention # SageAttention disabled
+    from scripts import build_sage_attention
 
     print_welcome_message()
     
-    # SageAttention build is temporarily disabled for faster iteration.
-    # print("--- [Setup] Attempting to build and install SageAttention... ---")
-    # try:
-    #     build_sage_attention.install_sage_attention()
-    #     print("--- [Setup] ✅ SageAttention installation process finished. ---")
-    # except Exception as e:
-    #     print(f"--- [Setup] ❌ SageAttention installation failed: {e}. Continuing with default attention. ---")
+    print("--- [Setup] Attempting to build and install SageAttention... ---")
+    try:
+        build_sage_attention.install_sage_attention()
+        print("--- [Setup] ✅ SageAttention installation process finished. ---")
+    except Exception as e:
+        print(f"--- [Setup] ❌ SageAttention installation failed: {e}. Continuing with default attention. ---")
 
 
     print("--- [Setup] Reloading site-packages to detect newly installed packages... ---")