From 18a06f8e5d14db7094558e485072d5a072585162 Mon Sep 17 00:00:00 2001 From: Andrina Zhang <78346223+andrinazxx@users.noreply.github.com> Date: Wed, 29 May 2024 00:01:18 -0700 Subject: [PATCH] Update Dockerfile - taming-transformers --- Dockerfile | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index d8a7f57..f60f8eb 100644 --- a/Dockerfile +++ b/Dockerfile @@ -21,7 +21,13 @@ USER jovyan # RUN conda install -y scikit-learn -RUN pip install --no-cache-dir networkx scipy librosa==0.10.0 wave soundfile lightning-utilities==0.7.1 pytorch-lightning==1.9.4 +RUN pip install --no-cache-dir networkx scipy librosa==0.10.0 wave soundfile pytorch-lightning==1.9.4 +RUN pip install --no-cache-dir lightning-utilities==0.7.1 + +# Clone and install the taming-transformers repository +RUN apt-get update && apt-get install -y git +RUN git clone https://github.com/CompVis/taming-transformers.git /taming-transformers +RUN pip install -e /taming-transformers # Override command to disable running jupyter notebook at launch # CMD ["/bin/bash"]