update prebuild

app.py

@@ -60,6 +60,8 @@ import gradio as gr
 from pathlib import Path
 from urllib.parse import urlparse
 from torch.hub import download_url_to_file, get_dir
+import shlex
+
 
 subprocess.run(
     "pip install flash-attn --no-build-isolation",
@@ -67,15 +69,6 @@ subprocess.run(
     shell=True,
 )
 
-
-result = subprocess.run(
-            ["bash", "setup.sh"],
-            capture_output=True,
-            text=True,
-            check=True
-        )
-print(f"✅ setup completed:\n{result.stdout}")
-
 def load_file_from_url(url, model_dir=None, progress=True, file_name=None):
     """Load file form http url, will download models if necessary.
 
@@ -117,7 +110,8 @@ pretrain_model_url = {
 	'vae': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/ema_vae.pth',
 	'dit': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/seedvr2_ema_3b.pth',
     'pos_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/pos_emb.pt',
-    'neg_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/neg_emb.pt'
+    'neg_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/neg_emb.pt',
+    'apex': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/apex-0.1-cp39-cp39-linux_x86_64.whl'
 }
 # download weights
 if not os.path.exists('./ckpts/seedvr2_ema_3b.pth'):
@@ -128,6 +122,11 @@ if not os.path.exists('./pos_emb.pt'):
 	load_file_from_url(url=pretrain_model_url['pos_emb'], model_dir='./', progress=True, file_name=None)
 if not os.path.exists('./neg_emb.pt'):
 	load_file_from_url(url=pretrain_model_url['neg_emb'], model_dir='./', progress=True, file_name=None)
+if not os.path.exists('./apex-0.1-cp39-cp39-linux_x86_64.whl'):
+	load_file_from_url(url=pretrain_model_url['apex'], model_dir='./', progress=True, file_name=None)
+
+subprocess.run(shlex.split("pip install apex-0.1-cp39-cp39-linux_x86_64.whl"))
+print(f"✅ setup completed Apex")
 
 # download images
 torch.hub.download_url_to_file(

setup.sh

@@ -1,8 +0,0 @@
-#!/bin/bash
-pip install --upgrade pip setuptools wheel
-
-git clone https://github.com/NVIDIA/apex
-cd apex
-# if pip >= 23.1 (ref: https://pip.pypa.io/en/stable/news/#v23-1) which supports multiple `--config-settings` with the same key... 
-pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./
-cd ..