update

app.py

@@ -111,7 +111,7 @@ pretrain_model_url = {
 	'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',
-    'apex': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/apex-0.1-cp39-cp39-linux_x86_64.whl'
+    'apex': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/apex-0.1-cp310-cp310-linux_x86_64.whl'
 }
 # download weights
 if not os.path.exists('./ckpts/seedvr2_ema_3b.pth'):
@@ -122,10 +122,10 @@ 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'):
+if not os.path.exists('./apex-0.1-cp310-cp310-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"))
+subprocess.run(shlex.split("pip install apex-0.1-cp310-cp310-linux_x86_64.whl"))
 print(f"✅ setup completed Apex")
 
 # download images

build_apex.sh

@@ -0,0 +1,45 @@
+#!/bin/bash
+set -e
+
+# 🧾 Config
+# APEX_COMMIT=f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0  # Known stable commit
+
+echo "🧹 Cleaning any previous apex build..."
+rm -rf apex
+rm -rf *.egg-info build dist
+
+echo "📥 Cloning NVIDIA/apex..."
+git clone https://github.com/NVIDIA/apex.git
+cd apex
+# git checkout $APEX_COMMIT
+
+echo "🛠 Installing build dependencies..."
+sudo apt-get update
+sudo apt-get install -y \
+  build-essential \
+  ninja-build \
+  python3-dev \
+  libffi-dev \
+  libncurses5-dev \
+  libncursesw5-dev \
+  libreadline-dev \
+  libssl-dev \
+  libsqlite3-dev \
+  zlib1g-dev \
+  libbz2-dev \
+  liblzma-dev \
+  git
+
+echo "🐍 Upgrading pip and wheel..."
+pip install -U pip setuptools wheel
+
+echo "🧪 Verifying PyTorch + CUDA availability..."
+python -c "import torch; print('PyTorch:', torch.__version__, '| CUDA:', torch.version.cuda)"
+
+echo "🔧 Building Apex with CUDA and C++ extensions..."
+python setup.py bdist_wheel --cuda_ext --cpp_ext
+
+echo "✅ Done! Built wheel:"
+ls -lh dist/*.whl
+
+cd ..