declare-lab/tango2

1FROM ubuntu:20.042LABEL maintainer="Hugging Face"3LABEL repository="diffusers"4 5ENV DEBIAN_FRONTEND=noninteractive6 7RUN apt update && \8    apt install -y bash \9                   build-essential \10                   git \11                   git-lfs \12                   curl \13                   ca-certificates \14                   libsndfile1-dev \15                   python3.8 \16                   python3-pip \17                   python3.8-venv && \18    rm -rf /var/lib/apt/lists19 20# make sure to use venv21RUN python3 -m venv /opt/venv22ENV PATH="/opt/venv/bin:$PATH"23 24# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)25# follow the instructions here: https://cloud.google.com/tpu/docs/run-in-container#train_a_jax_model_in_a_docker_container26RUN python3 -m pip install --no-cache-dir --upgrade pip && \27    python3 -m pip install --no-cache-dir \28        "jax[tpu]>=0.2.16,!=0.3.2" \29        -f https://storage.googleapis.com/jax-releases/libtpu_releases.html && \30    python3 -m pip install --upgrade --no-cache-dir \31        clu \32        "flax>=0.4.1" \33        "jaxlib>=0.1.65" && \34    python3 -m pip install --no-cache-dir \35        accelerate \36        datasets \37        hf-doc-builder \38        huggingface-hub \39        Jinja2 \40        librosa \        41        numpy \42        scipy \43        tensorboard \44        transformers45 46CMD ["/bin/bash"]