install cosmos-transfer1 in app.py instead of in requirements.txt

app.py

@@ -5,7 +5,8 @@ import gradio as gr
 import spaces
 
 PWD = os.path.dirname(__file__)
-CHECKPOINTS_PATH = "/data/checkpoints"
+# CHECKPOINTS_PATH = "/data/checkpoints"
+CHECKPOINTS_PATH = os.path.join(PWD, "checkpoints")
 
 # import subprocess
 
@@ -13,25 +14,32 @@ CHECKPOINTS_PATH = "/data/checkpoints"
 # subprocess.run("cp /usr/local/lib/python3.10/site-packages/nvidia/cudnn/include/*.h /usr/local/cuda/include", env={}, shell=True)
 # subprocess.run("cp /usr/local/lib/python3.10/site-packages/nvidia/cudnn/lib/*.so* /usr/local/cuda/lib64", env={}, shell=True)
 
-# setup env
-# os.environ["CUDA_HOME"] = "/usr/local/cuda"
-# os.environ["LD_LIBRARY_PATH"] = "$CUDA_HOME/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH"
-# os.environ["PATH"] = "$CUDA_HOME/bin:/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:$PATH"
-
 os.system("apt-get update && apt-get install -qqy libmagickwand-dev")
 
 # install packages
-os.system('export FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE && pip install "flash-attn<=2.7.4.post1" --no-build-isolation')
+os.system('export FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE && pip install --no-build-isolation "flash-attn<=2.7.4.post1"')
 os.system(
     "pip install https://download.pytorch.org/whl/cu128/flashinfer/flashinfer_python-0.2.5%2Bcu128torch2.7-cp38-abi3-linux_x86_64.whl"
 )
 os.system('export VLLM_ATTENTION_BACKEND=FLASHINFER && pip install "vllm==0.9.0"')
 os.system('pip install "decord==0.6.0"')
 
-os.system("ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/")
-os.system("ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/python3.10")
+os.system(
+    "export CONDA_PREFIX=/usr/local/cuda && ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/"
+)
+os.system(
+    "export CONDA_PREFIX=/usr/local/cuda && ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/python3.10"
+)
+
+os.system('pip install --no-build-isolation "transformer-engine[pytorch]"')
+os.system('pip install "decord==0.6.0"')
+
+os.system('pip install "git+https://github.com/nvidia-cosmos/cosmos-transfer1"')
 
-os.system('pip install "transformer-engine[pytorch]"')
+# setup env
+os.environ["CUDA_HOME"] = "/usr/local/cuda"
+os.environ["LD_LIBRARY_PATH"] = "$CUDA_HOME/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH"
+os.environ["PATH"] = "$CUDA_HOME/bin:/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:$PATH"
 
 from test_environment import main as check_environment
 

requirements.txt

@@ -3,8 +3,48 @@ wheel
 ninja
 
 # cosmos-transfer1
-git+https://github.com/nvidia-cosmos/cosmos-transfer1
+# git+https://github.com/nvidia-cosmos/cosmos-transfer1
 
 # other packages
-https://download.pytorch.org/whl/cu128/flashinfer/flashinfer_python-0.2.5%2Bcu128torch2.7-cp38-abi3-linux_x86_64.whl
+# https://download.pytorch.org/whl/cu128/flashinfer/flashinfer_python-0.2.5%2Bcu128torch2.7-cp38-abi3-linux_x86_64.whl
+# decord==0.6.0
+
+apex==0.9.10dev
+attr==0.3.2
+attrs==25.3.0
+better_profanity==0.7.0
+cloudpickle==3.1.1
 decord==0.6.0
+dill==0.4.0
+einops==0.8.1
+hydra-core==1.3.2
+imageio[pyav,ffmpeg]==2.37.0
+iopath==0.1.10
+loguru==0.7.3
+matplotlib==3.10.3
+mediapy==1.2.4
+megatron_core
+natsort==8.4.0
+nltk==3.9.1
+numpy==2.2.6
+nvidia_ml_py==12.570.86
+omegaconf==2.3.0
+opencv_python==4.10.0.84
+opencv_python_headless==4.11.0.86
+packaging==25.0
+pandas==2.2.3
+peft==0.14.0
+Pillow==11.2.1
+pillow
+pycocotools==2.0.8
+pynvml==12.0.0
+python-magic
+PyYAML==6.0.2
+retinaface_py==0.0.2
+rtmlib==0.0.13
+sam2==1.1.0
+termcolor==3.1.0
+torch==2.7.0
+torchvision==0.22.0
+tqdm==4.67.1
+transformers==4.49.0