Upload folder using huggingface_hub

scripts/build_sage_attention.py

@@ -1,10 +1,9 @@
 import os
 import subprocess
 import sys
-import textwrap
+import importlib.util
 
-REPO_URL = "https://github.com/thu-ml/SageAttention.git"
-REPO_DIR = "SageAttention"
+WHEEL_URL = "https://huggingface.co/RioShiina/Sage-Attention-ZeroGPU-Space-Build/resolve/main/sageattention-2.2.0-cp310-cp310-linux_x86_64.whl?download=true"
 
 def run_command(command, cwd=None, env=None):
     print(f"🚀 Running command: {' '.join(command)}")
@@ -17,83 +16,42 @@ def run_command(command, cwd=None, env=None):
         text=True
     )
     
-    if result.returncode != 0:
+    if result.stdout:
         print(result.stdout)
-        raise subprocess.CalledProcessError(result.returncode, command)
-
-def patch_setup_py(setup_py_path):
-    print(f"--- [SageAttention Build] Applying patches to {setup_py_path} ---")
-
-    with open(setup_py_path, 'r', encoding='utf-8') as f:
-        content = f.read()
-
-    original_cxx_flags = 'CXX_FLAGS = ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17", "-DENABLE_BF16"]'
-    modified_cxx_flags = 'CXX_FLAGS = ["-g", "-O3", "-std=c++17", "-DENABLE_BF16"]'
-    
-    if original_cxx_flags in content:
-        content = content.replace(original_cxx_flags, modified_cxx_flags)
-        print("🔧 Patch 1/1: Removed '-fopenmp' and '-lgomp' from CXX_FLAGS.")
-    else:
-        print("⚠️ Patch 1/1: CXX_FLAGS line not found as expected. It might have been changed upstream. Skipping.")
-
-    with open(setup_py_path, 'w', encoding='utf-8') as f:
-        f.write(content)
     
-    print("✅ Patches applied successfully.")
+    if result.returncode != 0:
+        raise subprocess.CalledProcessError(result.returncode, command)
 
+def is_sage_installed():
+    return importlib.util.find_spec("sageattention") is not None
 
 def install_sage_attention():
-    print("--- [SageAttention Build] Checking environment ---")
+    package_name = "sageattention"
+    print(f"--- [SageAttention Install] Checking for {package_name} ---")
     
-    if os.path.isdir(REPO_DIR):
-        print(f"✅ Directory '{REPO_DIR}' already exists, assuming SageAttention is installed. Skipping build.")
+    if is_sage_installed():
+        print(f"✅ {package_name} package is already installed. Skipping installation.")
         return
 
-    print(f"⏳ Directory '{REPO_DIR}' not found. Starting a fresh installation of SageAttention.")
+    print(f"⏳ {package_name} package not found. Starting installation from pre-built wheel.")
     
     try:
-        print(f"--- [SageAttention Build] Step 1/3: Cloning repository ---")
-        run_command(["git", "clone", REPO_URL])
-        print("✅ Repository cloned successfully.")
-
-        print(f"--- [SageAttention Build] Step 2/3: Patching setup.py ---")
-        setup_py_path = os.path.join(REPO_DIR, "setup.py")
-        patch_setup_py(setup_py_path)
-
-        print(f"--- [SageAttention Build] Step 3/3: Compiling and installing ---")
+        print(f"--- [SageAttention Install] Installing from URL: {WHEEL_URL} ---")
         
-        build_env = os.environ.copy()
-        build_env.update({
-            "TORCH_CUDA_ARCH_LIST": "9.0",
-            "EXT_PARALLEL": "4",
-            "NVCC_APPEND_FLAGS": "--threads 8",
-            "MAX_JOBS": "32"
-        })
-        print("🔧 Setting build environment variables:")
-        print(f"   - TORCH_CUDA_ARCH_LIST='{build_env['TORCH_CUDA_ARCH_LIST']}'")
-        print(f"   - EXT_PARALLEL={build_env['EXT_PARALLEL']}")
-        print(f"   - NVCC_APPEND_FLAGS='{build_env['NVCC_APPEND_FLAGS']}'")
-        print(f"   - MAX_JOBS={build_env['MAX_JOBS']}")
-
-        install_command = [sys.executable, "setup.py", "install"]
+        pip_command = [sys.executable, "-m", "pip", "install", WHEEL_URL]
         
-        run_command(install_command, cwd=REPO_DIR, env=build_env)
+        run_command(pip_command)
 
-        print("🎉 SageAttention compiled and installed successfully! ---")
+        print("🎉 SageAttention installed successfully from pre-built wheel! ---")
 
-    except FileNotFoundError:
-        print("❌ ERROR: 'git' command not found. Please ensure Git is installed in your environment.")
-        sys.exit(1)
     except subprocess.CalledProcessError as e:
         print(f"❌ Command failed with return code: {e.returncode}")
         print(f"❌ Command: {' '.join(e.cmd)}")
         print("❌ SageAttention installation failed. Please check the logs above for details.")
-        sys.exit(1)
+        raise e
     except Exception as e:
-        print(f"❌ An unknown error occurred: {e}")
-        sys.exit(1)
+        print(f"❌ An unknown error occurred during installation: {e}")
+        raise e
 
 if __name__ == "__main__":
-    if os.path.isdir(REPO_DIR):
-         print(f"Note: To force a rebuild, please delete the '{REPO_DIR}' directory first.")
     install_sage_attention()

yaml/file_list.yaml

@@ -416,10 +416,6 @@ file:
       source: "hf"
       repo_id: "duongve/AnimaYume"
       repository_file_path: "split_files/diffusion_models/AnimaYume_tuned_v04.safetensors"
-    - filename: "anima_pencil-v1.0.safetensors"
-      source: "hf"
-      repo_id: "bluepen5805/anima-models"
-      repository_file_path: "anima_pencil-v1.0.safetensors"
     # NewBie-Image
     - filename: "NewBie-Image-Exp0.1-bf16.safetensors"
       source: "hf"

yaml/model_list.yaml

@@ -97,11 +97,6 @@ Checkpoint:
           unet: "AnimaYume_tuned_v04.safetensors"
           vae: "qwen_image_vae.safetensors"
           clip: "qwen_3_06b_base.safetensors"
-      - display_name: "bluepen5805/Anima-pencil-v1.0"
-        components:
-          unet: "anima_pencil-v1.0.safetensors"
-          vae: "qwen_image_vae.safetensors"
-          clip: "qwen_3_06b_base.safetensors"
       - display_name: "circlestone-labs/Anima-base-v1.0"
         components:
           unet: "anima-base-v1.0.safetensors"