update

Dockerfile

@@ -29,8 +29,8 @@ USER user
 
 RUN conda create -n gradio-user python=3.11
 RUN conda run -n gradio-user pip install --upgrade pip
-RUN conda install -n gradio-user pytorch3d=0.7.7 -c pytorch3d -c conda-forge
-RUN conda install -n gradio-user -c conda-forge open-clip-torch pytorch-lightning
+# RUN conda install -n gradio-user pytorch3d=0.7.7 -c pytorch3d -c conda-forge
+# RUN conda install -n gradio-user -c conda-forge open-clip-torch pytorch-lightning
 RUN conda run -n gradio-user pip install diffusers transformers accelerate xformers controlnet-aux gradio trimesh xatlas scikit-learn opencv-python omegaconf
 
 # Set home to the user's home directory

run.sh

@@ -4,6 +4,9 @@ CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
 eval "$(conda shell.bash hook)"
 conda activate gradio-user
 
+conda install -n gradio-user pytorch3d=0.7.7 -c pytorch3d -c conda-forge
+conda install -n gradio-user -c conda-forge open-clip-torch pytorch-lightning
+
 # Start app.py
 echo "Starting app.py..."
 export OMP_NUM_THREADS=4 # default is a wrong value: 7500m