Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions .dockerignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
# The .dockerignore file excludes files from the container build process.
#
# https://docs.docker.com/engine/reference/builder/#dockerignore-file

# Exclude Git files
.git
.github
.gitignore

# Exclude Python cache files
__pycache__
.mypy_cache
.pytest_cache
.ruff_cache

# Exclude Python virtual environment
/venv

2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@ Audio samples: [https://styletts2.github.io/](https://styletts2.github.io/)

Online demo: [Hugging Face](https://huggingface.co/spaces/styletts2/styletts2) (thank [@fakerybakery](https://github.com/fakerybakery) for the wonderful online demo)

Online demo: [Replicate](https://replicate.com/adirik/styletts2) (demos on Replicate shut down after a while of inactivity, cold starts take 3-5 minutes.)

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/yl4579/StyleTTS2/blob/main/) [![Slack](https://img.shields.io/badge/Join%20Our%20Community-Slack-blue)](https://join.slack.com/t/styletts2/shared_invite/zt-2805io6cg-0ROMhjfW9Gd_ix_FJqjGmQ)

## TODO
Expand Down
33 changes: 33 additions & 0 deletions cog.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

build:
cuda: "12.1"
gpu: true
system_packages:
- "espeak-ng"
- "ffmpeg"
python_version: "3.9"
python_packages:
- "numpy==1.26.2"
- "torch==2.1.1"
- "torchvision==0.16.1"
- "SoundFile==0.12.1"
- "torchaudio==2.1.1"
- "munch==4.0.0"
- "pydub"
- "librosa==0.10.1"
- "nltk==3.8.1"
- "matplotlib"
- "accelerate==0.24.1"
- "transformers==4.35.2"
- "einops==0.7.0"
- "einops-exts"
- "tqdm==4.66.1"
- "phonemizer==3.2.1"
- "typing"
- "typing-extensions"
- "ipython==8.17.2"
- "git+https://github.com/resemble-ai/monotonic_align.git"
predict: "predict.py:Predictor"

Loading