[sapiens] Fixed lite

Author: fzoric8

sapiens/README.md

@@ -18,6 +18,8 @@ This Dockerfile enables plug&play start for the sapiens foundational models for
 
 Folder lite contains docker that builds environment only for the sapiens inference. 
 
+Weights for the sapiens lite can be found [here](https://huggingface.co/facebook/sapiens/tree/main/sapiens_lite_host). 
+
 ### Full 
 
 Folder full contains Dockerfile that builds environment for the training (fine-tuning) env for sapiens. 

sapiens/lite/Dockerfile

@@ -13,9 +13,9 @@ RUN apt-get update -y && \
 WORKDIR /root
 
 # Conda
-#ENV CONDA_DIR /opt/conda
-#RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
-#    /bin/bash ~/miniconda.sh -b -p /opt/conda
+ENV CONDA_DIR /opt/conda
+RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
+    /bin/bash ~/miniconda.sh -b -p /opt/conda
 
 # Put conda in path so we can use conda activate
 ENV PATH=$CONDA_DIR/bin:$PATH
@@ -29,7 +29,7 @@ ENV SAPIENS_LITE_ROOT=$SAPIENS_ROOT/lite
 
 # Conda in docker: https://medium.com/@chadlagore/conda-environments-with-docker-82cdc9d25754
 # Conda in docker: https://pythonspeed.com/articles/activate-conda-dockerfile/
-# Minimal sapiens_lite conda env 
+# Minimal sapiens_lite conda env
 RUN conda init bash \
     && . ~/.bashrc \
     && conda create --name sapiens_lite python=3.10 \

sapiens/lite/first_run.sh

@@ -0,0 +1,25 @@
+#!/bin/bash
+
+xhost local:root
+
+CONTAINER_NAME=sapiens_lite_cont
+IMAGE_NAME=sapiens_img:lite
+
+# Hook to the current SSH_AUTH_LOCK - since it changes
+# https://www.talkingquickly.co.uk/2021/01/tmux-ssh-agent-forwarding-vs-code/
+ln -sf $SSH_AUTH_SOCK ~/.ssh/ssh_auth_sock
+
+docker run \
+  -it \
+  --network host \
+  --privileged \
+  --gpus all \
+  --volume /dev:/dev \
+  --volume /tmp/.x11-unix:/tmp/.x11-unix \
+  --volume ~/.ssh/ssh_auth_sock:/ssh-agent \
+  --env SSH_AUTH_SOCK=/ssh-agent \
+  --env DISPLAY=$DISPLAY \
+  --env TERM=xterm-256color \
+  --name $CONTAINER_NAME \
+  $IMAGE_NAME \
+  /bin/bash