Skip to content

Commit

Permalink
add xformers install step, fix Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
rsxdalv committed Nov 23, 2024
1 parent b09905a commit b5ea01e
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 11 deletions.
5 changes: 5 additions & 0 deletions .github/workflows/test-pip-install.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,10 @@ jobs:
if: matrix.platform == 'macos-14'
run: |
python -m pip install https://github.com/rsxdalv/xformers/releases/download/v0.0.27/xformers-0.0.27+184b280.d20241026-cp310-cp310-macosx_11_0_universal2.whl
- name: Install xformers non-macos
if: matrix.platform != 'macos-14'
run: |
python -m pip install xformers==0.0.27+cu118 --index-url https://download.pytorch.org/whl/cu118
- name: Install dependencies
# python -m pip install --upgrade pip==23.3.2
# python -m pip install -r requirements.txt
Expand All @@ -51,6 +55,7 @@ jobs:
# python -m pip install hydra-core==1.3.2
# python -m pip install nvidia-ml-py
run: |
python -m pip install
python -m pip install -r requirements.txt -r requirements_bark_hubert_quantizer.txt -r requirements_rvc.txt -r requirements_audiocraft.txt -r requirements_styletts2.txt -r requirements_vall_e.txt -r requirements_maha_tts.txt -r requirements_stable_audio.txt hydra-core==1.3.2 nvidia-ml-py
# - name: Lint with ruff
Expand Down
22 changes: 11 additions & 11 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ RUN npm --version
ENV TORCH_VERSION=2.3.1

ENV PATH="/root/.cargo/bin:$PATH"
RUN pip install --no-cache-dir --system setuptools torch==$TORCH_VERSION torchvision torchaudio
RUN pip install --no-cache-dir setuptools torch==$TORCH_VERSION torchvision torchaudio

# Set working directory
WORKDIR /app
Expand All @@ -39,16 +39,16 @@ RUN pip3 install --no-cache-dir --upgrade pip==23.3.2 setuptools

# Install all requirements
RUN pip3 install --no-cache-dir torch==$TORCH_VERSION -r requirements.txt
# RUN pip install --no-cache-dir --system --verbose torch==$TORCH_VERSION -r requirements.txt
RUN pip install --no-cache-dir --system torch==$TORCH_VERSION -r requirements_bark_hubert_quantizer.txt
RUN pip install --no-cache-dir --system torch==$TORCH_VERSION -r requirements_rvc.txt
RUN pip install --no-cache-dir --system torch==$TORCH_VERSION -r requirements_audiocraft.txt
RUN pip install --no-cache-dir --system torch==$TORCH_VERSION -r requirements_styletts2.txt
RUN pip install --no-cache-dir --system torch==$TORCH_VERSION -r requirements_vall_e.txt
RUN pip install --no-cache-dir --system torch==$TORCH_VERSION -r requirements_maha_tts.txt
RUN pip install --no-cache-dir --system torch==$TORCH_VERSION -r requirements_stable_audio.txt
# RUN pip install --no-cache-dir --system torch==$TORCH_VERSION hydra-core==1.3.2
RUN pip install --no-cache-dir --system torch==$TORCH_VERSION nvidia-ml-py
# RUN pip install --no-cache-dir --verbose torch==$TORCH_VERSION -r requirements.txt
RUN pip install --no-cache-dir torch==$TORCH_VERSION -r requirements_bark_hubert_quantizer.txt
RUN pip install --no-cache-dir torch==$TORCH_VERSION -r requirements_rvc.txt
RUN pip install --no-cache-dir torch==$TORCH_VERSION -r requirements_audiocraft.txt
RUN pip install --no-cache-dir torch==$TORCH_VERSION -r requirements_styletts2.txt
RUN pip install --no-cache-dir torch==$TORCH_VERSION -r requirements_vall_e.txt
RUN pip install --no-cache-dir torch==$TORCH_VERSION -r requirements_maha_tts.txt
RUN pip install --no-cache-dir torch==$TORCH_VERSION -r requirements_stable_audio.txt
# RUN pip install --no-cache-dir torch==$TORCH_VERSION hydra-core==1.3.2
RUN pip install --no-cache-dir torch==$TORCH_VERSION nvidia-ml-py


# add postgres & run setup
Expand Down

0 comments on commit b5ea01e

Please sign in to comment.