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 --upgrade --no-cache-dir \28        clu \29        "jax[cpu]>=0.2.16,!=0.3.2" \30        "flax>=0.4.1" \31        "jaxlib>=0.1.65" && \32    python3 -m pip install --no-cache-dir \33        accelerate \34        datasets \35        hf-doc-builder \36        huggingface-hub \37        Jinja2 \38        librosa \39        numpy \40        scipy \41        tensorboard \42        transformers43 44CMD ["/bin/bash"]