reset

requirements.txt

@@ -1,11 +1,48 @@
 --extra-index-url https://download.pytorch.org/whl/cu124
-git+https://github.com/huggingface/diffusers.git@main
-git+https://github.com/huggingface/transformers.git@main
-torch>=2.4.0,<2.6.0
-gradio>=4.0.0
-safetensors
+bitsandbytes
+decord
+einops
+facexlib
+ftfy
+gguf
+git+https://github.com/huggingface/accelerate.git@main#egg=accelerate
+git+https://github.com/huggingface/diffusers.git@main#egg=diffusers
+git+https://github.com/huggingface/transformers.git@main#egg=transformers
+gradio
+hf_transfer
 huggingface_hub
 imageio
-opencv-python-headless
-Pillow>=10.2.0
+imageio-ffmpeg
+insightface
+invisible_watermark
+matplotlib
+moviepy==1.0.3
 numpy<2.0
+onnxruntime
+onnxruntime-gpu
+omegaconf
+opencv-python
+opencv-python-headless
+git+https://github.com/huggingface/optimum-quanto
+packaging
+patch_conv
+Pillow==10.2.0
+psutil
+safetensors
+scipy
+scikit-learn
+scikit-image
+scikit-video
+sentencepiece
+setuptools
+spaces
+timm
+tokenizers>=0.13.3
+torch<2.6.0,>=2.4.0
+torchao
+torchaudio
+torchsde
+torchvision
+tqdm
+wheel
+git+https://github.com/huggingface/peft.git

utils.py

@@ -1,9 +1,40 @@
 def install_packages():
     import subprocess
     import sys
+    import importlib
 
-    subprocess.run([
-        sys.executable, "-m", "pip", "install",
-        "-r", "requirements.txt",
-        "--upgrade"
-    ], check=True)
+    def _is_package_available(name) -> bool:
+        try:
+            importlib.import_module(name)
+            return True
+        except (ImportError, ModuleNotFoundError):
+            return False
+
+    # upgrade pip
+    subprocess.run(
+        f"{sys.executable} -m pip install --upgrade pip", shell=True, check=True
+    )
+    subprocess.run(
+        f"{sys.executable} -m pip install --upgrade ninja wheel setuptools packaging", shell=True, check=True
+    )
+
+    # install ninja
+    if not _is_package_available("ninja"):
+        subprocess.run(f"{sys.executable} -m pip install ninja nvidia-cudnn-cu12==9.1.0.70 nvidia-cublas-cu12==12.4.5.8 torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cu124", shell=True, check=True)
+
+    # install flash attention
+    if not _is_package_available("flash_attn"):
+        subprocess.run(
+            f"{sys.executable} -m pip install -v -U flash-attention --no-build-isolation",
+            env={"MAX_JOBS": "1"},
+            shell=True,
+            check=True
+        )
+
+    # install xformers
+    if not _is_package_available("xformers"):
+        subprocess.run(
+            f"{sys.executable} -m pip install -v -U xformers nvidia-cudnn-cu12==9.1.0.70 nvidia-cublas-cu12==12.4.5.8 torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cu124",
+            shell=True,
+            check=True
+        )