Upload folder using huggingface_hub

app.py

@@ -46,17 +46,9 @@ def dummy_gpu_for_startup():
 
 
 def main():
-    from scripts import build_sage_attention
     from comfy_integration import setup as setup_comfyui
     from utils.app_utils import load_ipadapter_presets
     
-    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] Starting ComfyUI initialization ---")
     setup_comfyui.initialize_comfyui()
     

core/pipelines/workflow_recipes/_partials/conditioning/hidream.yaml

@@ -1,53 +0,0 @@
-nodes:
-  unet_loader:
-    class_type: UNETLoader
-    title: "Load HiDream UNET"
-    params:
-      weight_dtype: "default"
-  vae_loader:
-    class_type: VAELoader
-    title: "Load HiDream VAE"
-  clip_loader:
-    class_type: QuadrupleCLIPLoader
-    title: "Load HiDream Quadruple CLIP"
-  
-  model_sampler:
-    class_type: ModelSamplingSD3
-    title: "ModelSamplingSD3"
-    params:
-      shift: 6.0
-
-connections:
-  - from: "unet_loader:0"
-    to: "model_sampler:model"
-    
-  - from: "model_sampler:0"
-    to: "ksampler:model"
-    
-  - from: "clip_loader:0"
-    to: "pos_prompt:clip"
-  - from: "clip_loader:0"
-    to: "neg_prompt:clip"
-
-  - from: "pos_prompt:0"
-    to: "ksampler:positive"
-  - from: "neg_prompt:0"
-    to: "ksampler:negative"
-    
-  - from: "vae_loader:0"
-    to: "vae_decode:vae"
-  - from: "vae_loader:0"
-    to: "vae_encode:vae"
-
-dynamic_conditioning_chains:
-  conditioning_chain:
-    ksampler_node: "ksampler"
-    clip_source: "clip_loader:0"
-
-ui_map:
-  unet_name: "unet_loader:unet_name"
-  vae_name: "vae_loader:vae_name"
-  clip1_name: "clip_loader:clip_name1"
-  clip2_name: "clip_loader:clip_name2"
-  clip3_name: "clip_loader:clip_name3"
-  clip4_name: "clip_loader:clip_name4"

requirements.txt

@@ -37,26 +37,10 @@ PyOpenGL
 glfw
 
 
-addict
-albumentations
-ftfy
-fvcore
+diffusers
+protobuf
 huggingface-hub
-imageio[ffmpeg]
-importlib_metadata
-matplotlib
-mediapipe
-ninja
-omegaconf
-opencv-python>=4.7.0.72
-python-dateutil
-scikit-image
-scikit-learn
-soundfile
+imageio
+
 spaces
-svglib
-trimesh[easy]
-yacs
-yapf
-onnxruntime-gpu
-diffusers
+sageattention @ https://huggingface.co/RioShiina/Sage-Attention-ZeroGPU-Space-Build/resolve/main/sageattention-2.2.0-cp310-cp310-linux_x86_64.whl

scripts/build_sage_attention.py

@@ -1,57 +0,0 @@
-import os
-import subprocess
-import sys
-import importlib.util
-
-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)}")
-    result = subprocess.run(
-        command,
-        cwd=cwd,
-        env=env,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT,
-        text=True
-    )
-    
-    if result.stdout:
-        print(result.stdout)
-    
-    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():
-    package_name = "sageattention"
-    print(f"--- [SageAttention Install] Checking for {package_name} ---")
-    
-    if is_sage_installed():
-        print(f"✅ {package_name} package is already installed. Skipping installation.")
-        return
-
-    print(f"⏳ {package_name} package not found. Starting installation from pre-built wheel.")
-    
-    try:
-        print(f"--- [SageAttention Install] Installing from URL: {WHEEL_URL} ---")
-        
-        pip_command = [sys.executable, "-m", "pip", "install", WHEEL_URL]
-        
-        run_command(pip_command)
-
-        print("🎉 SageAttention installed successfully from pre-built wheel! ---")
-
-    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.")
-        raise e
-    except Exception as e:
-        print(f"❌ An unknown error occurred during installation: {e}")
-        raise e
-
-if __name__ == "__main__":
-    install_sage_attention()