diff --git a/Chatterbox_TTS_Colab_Demo.ipynb b/Chatterbox_TTS_Colab_Demo.ipynb index cd8ef79..1f8af16 100644 --- a/Chatterbox_TTS_Colab_Demo.ipynb +++ b/Chatterbox_TTS_Colab_Demo.ipynb @@ -41,7 +41,7 @@ }, { "cell_type": "code", - "source": "%%bash\nset -euo pipefail\n\ncd /content\nMICROMAMBA=\"/content/bin/micromamba\"\n\nts() { date +\"[%Y-%m-%d %H:%M:%S]\"; }\n\necho \"$(ts) Sanity check micromamba:\"\nls -lah \"$MICROMAMBA\"\n\necho \"$(ts) Upgrading pip tooling inside cb311...\"\n\"$MICROMAMBA\" run -n cb311 python -m pip install -U pip setuptools wheel --progress-bar on\n\necho \"$(ts) Installing PyTorch 2.5.1 (CUDA 12.1)... (this can take a while)\"\n\"$MICROMAMBA\" run -n cb311 pip install \\\n --progress-bar on \\\n torch==2.5.1+cu121 torchaudio==2.5.1+cu121 torchvision==0.20.1+cu121 \\\n --index-url https://download.pytorch.org/whl/cu121\n\necho \"$(ts) Installing Chatterbox package (from GitHub, no-cache, upgrade)...\"\n\"$MICROMAMBA\" run -n cb311 pip uninstall -y chatterbox-tts chatterbox || true\n\"$MICROMAMBA\" run -n cb311 pip install \\\n --no-cache-dir --upgrade \\\n --progress-bar on \\\n \"chatterbox-tts @ git+https://github.com/devnen/chatterbox-v2.git@master\"\n\necho \"$(ts) Installing s3tokenizer + onnx (--no-deps to avoid protobuf conflict)...\"\n\"$MICROMAMBA\" run -n cb311 pip install --no-deps s3tokenizer==0.3.0 onnx==1.16.0\n\necho \"$(ts) Force-upgrading protobuf for onnx compatibility...\"\n\"$MICROMAMBA\" run -n cb311 pip install --no-deps --force-reinstall \"protobuf>=4.25.0\"\n\necho \"$(ts) ✅ Installation complete!\"", + "source": "%%bash\nset -euo pipefail\n\ncd /content\nMICROMAMBA=\"/content/bin/micromamba\"\n\nts() { date +\"[%Y-%m-%d %H:%M:%S]\"; }\n\necho \"$(ts) Sanity check micromamba:\"\nls -lah \"$MICROMAMBA\"\n\necho \"$(ts) Upgrading pip tooling inside cb311...\"\n\"$MICROMAMBA\" run -n cb311 python -m pip install -U pip setuptools wheel --progress-bar on\n\necho \"$(ts) Installing PyTorch 2.5.1 (CUDA 12.1)... (this can take a while)\"\n\"$MICROMAMBA\" run -n cb311 pip install \\\n --progress-bar on \\\n torch==2.5.1+cu121 torchaudio==2.5.1+cu121 torchvision==0.20.1+cu121 \\\n --index-url https://download.pytorch.org/whl/cu121\n\necho \"$(ts) Installing Chatterbox package (from GitHub, no-cache, upgrade)...\"\n\"$MICROMAMBA\" run -n cb311 pip uninstall -y chatterbox-tts chatterbox || true\n\"$MICROMAMBA\" run -n cb311 pip install \\\n --no-cache-dir --upgrade \\\n --progress-bar on \\\n \"chatterbox-tts @ git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2\"\n\necho \"$(ts) Installing s3tokenizer + onnx (--no-deps to avoid protobuf conflict)...\"\n\"$MICROMAMBA\" run -n cb311 pip install --no-deps s3tokenizer==0.3.0 onnx==1.16.0\n\necho \"$(ts) Force-upgrading protobuf for onnx compatibility...\"\n\"$MICROMAMBA\" run -n cb311 pip install --no-deps --force-reinstall \"protobuf>=4.25.0\"\n\necho \"$(ts) ✅ Installation complete!\"", "metadata": { "id": "2alxTXqZuGhl" }, diff --git a/Dockerfile b/Dockerfile index 763dafe..b31bc44 100644 --- a/Dockerfile +++ b/Dockerfile @@ -42,7 +42,7 @@ RUN pip3 install --no-cache-dir --upgrade pip && \ if [ "$RUNTIME" = "nvidia" ]; then \ pip3 install --no-cache-dir -r requirements-nvidia.txt; \ fi && \ - pip3 install --no-cache-dir --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 && \ + pip3 install --no-cache-dir --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 && \ pip3 install --no-cache-dir "protobuf>=4.25.0" # Copy the rest of the application code COPY . . diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 879f31e..bdb287e 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -27,7 +27,7 @@ COPY requirements.txt ./requirements.txt # 2. Chatterbox with --no-deps to prevent pip from pulling conflicting versions RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir -r requirements.txt && \ - python3 -m pip install --no-cache-dir --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 && \ + python3 -m pip install --no-cache-dir --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 && \ python3 -m pip install --no-cache-dir "protobuf>=4.25.0" # Copy the rest of the application code diff --git a/Dockerfile.cu128 b/Dockerfile.cu128 index c0f38d7..4e2f92a 100644 --- a/Dockerfile.cu128 +++ b/Dockerfile.cu128 @@ -35,7 +35,7 @@ COPY requirements-nvidia-cu128.txt ./requirements-nvidia-cu128.txt # 2. Chatterbox with --no-deps to prevent pip from downgrading torch RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir -r requirements-nvidia-cu128.txt && \ - python3 -m pip install --no-cache-dir --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 && \ + python3 -m pip install --no-cache-dir --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 && \ python3 -m pip install --no-cache-dir "protobuf>=4.25.0" # Copy the rest of the application code diff --git a/Dockerfile.cu130 b/Dockerfile.cu130 index 0c976b0..3ea95a8 100644 --- a/Dockerfile.cu130 +++ b/Dockerfile.cu130 @@ -35,7 +35,7 @@ COPY requirements-nvidia-cu130.txt ./requirements-nvidia-cu130.txt # 2. Chatterbox with --no-deps to prevent pip from downgrading torch RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir -r requirements-nvidia-cu130.txt && \ - python3 -m pip install --no-cache-dir --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 && \ + python3 -m pip install --no-cache-dir --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 && \ python3 -m pip install --no-cache-dir "protobuf>=4.25.0" # Copy the rest of the application code diff --git a/Dockerfile.rdna4 b/Dockerfile.rdna4 index 16400b8..e84e6d8 100644 --- a/Dockerfile.rdna4 +++ b/Dockerfile.rdna4 @@ -31,7 +31,7 @@ COPY requirements-rdna4.txt ./requirements-rdna4.txt RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir -r requirements-rdna4-init.txt && \ python3 -m pip install --no-cache-dir -r requirements-rdna4.txt && \ - python3 -m pip install --no-cache-dir --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 && \ + python3 -m pip install --no-cache-dir --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 && \ python3 -m pip install --no-cache-dir "protobuf>=4.25.0" # Patches for torch 2.9+ compatibility (required when using ROCm 7.2 wheels): diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 7490204..779455c 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -37,7 +37,7 @@ COPY requirements-rocm.txt ./requirements-rocm.txt RUN python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir -r requirements-rocm-init.txt && \ python3 -m pip install --no-cache-dir -r requirements-rocm.txt && \ - python3 -m pip install --no-cache-dir --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 && \ + python3 -m pip install --no-cache-dir --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 && \ python3 -m pip install --no-cache-dir "protobuf>=4.25.0" # Copy the rest of the application code diff --git a/README.md b/README.md index f31e77b..8d9df0a 100644 --- a/README.md +++ b/README.md @@ -505,7 +505,7 @@ This is the most straightforward option and works on any machine without a compa # Make sure your (venv) is active pip install --upgrade pip pip install -r requirements.txt -pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master +pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 ```
@@ -525,7 +525,7 @@ For users with NVIDIA GPUs. This provides the best performance for RTX 20/30/40 # Make sure your (venv) is active pip install --upgrade pip pip install -r requirements-nvidia.txt -pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master +pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 ``` **After installation, verify that PyTorch can see your GPU:** @@ -568,7 +568,7 @@ pip install --upgrade pip pip install -r requirements-nvidia-cu128.txt # Step 2: Install chatterbox without dependencies (prevents PyTorch downgrade) -pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master +pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 ``` ⚠️ **Critical:** The `--no-deps` flag is required to prevent PyTorch from being downgraded to a version that doesn't support Blackwell GPUs. @@ -628,7 +628,7 @@ pip install -r requirements-rocm-init.txt pip install -r requirements-rocm.txt # Step 3: Install chatterbox without dependencies (prevents ROCm torch overwrite) -pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master +pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 ``` ⚠️ **Critical:** The `--no-deps` flag on chatterbox-tts is required to prevent pip from replacing the ROCm PyTorch wheels with CPU-only versions from PyPI. The `start.py` launcher handles this automatically. @@ -675,7 +675,7 @@ tts_engine: **Step 3: Install remaining dependencies** ```bash # Install chatterbox-tts without its dependencies to avoid conflicts -pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master +pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 # Install core server dependencies pip install fastapi 'uvicorn[standard]' librosa safetensors soundfile pydub audiotsm praat-parselmouth python-multipart requests aiofiles PyYAML watchdog unidecode inflect tqdm @@ -1036,7 +1036,7 @@ After you have updated the code using Method 2 or 3, complete these final steps. * **For NVIDIA GPU Systems (CUDA 12.8 / Blackwell):** ```bash pip install -r requirements-nvidia-cu128.txt - pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master + pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 ``` * **For AMD GPU Systems:** ```bash diff --git a/README_CUDA128.md b/README_CUDA128.md index a2867d0..4be5db6 100644 --- a/README_CUDA128.md +++ b/README_CUDA128.md @@ -118,7 +118,7 @@ pip install -r requirements-nvidia-cu128.txt # IMPORTANT: Install Chatterbox separately with --no-deps # This prevents PyTorch from being downgraded -pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master +pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 # Start the server python server.py @@ -201,7 +201,7 @@ This error means PyTorch doesn't support your GPU's compute capability. This typ # Or manually pip install torch==2.9.0 torchvision==0.24.0 torchaudio==2.9.0 --index-url https://download.pytorch.org/whl/cu128 - pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master + pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 ``` 3. **Check supported architectures**: diff --git a/README_Colab.md b/README_Colab.md index 193c6be..203e748 100644 --- a/README_Colab.md +++ b/README_Colab.md @@ -79,7 +79,7 @@ echo "$(ts) Installing Chatterbox package (from GitHub, no-cache, upgrade)..." "$MICROMAMBA" run -n cb311 pip install \ --no-cache-dir --upgrade \ --progress-bar on \ - "chatterbox-tts @ git+https://github.com/devnen/chatterbox-v2.git@master" + "chatterbox-tts @ git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2" echo "$(ts) ✅ Installation complete!" ``` diff --git a/requirements-nvidia-cu128.txt b/requirements-nvidia-cu128.txt index ec5c841..a1d6006 100644 --- a/requirements-nvidia-cu128.txt +++ b/requirements-nvidia-cu128.txt @@ -14,7 +14,7 @@ # # For manual installation, run these commands in order: # pip install -r requirements-nvidia-cu128.txt -# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 +# pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 # ============================================================================ # --- PyTorch CUDA 12.8 Package Index --- diff --git a/requirements-nvidia-cu130.txt b/requirements-nvidia-cu130.txt index b6e4820..47809ca 100644 --- a/requirements-nvidia-cu130.txt +++ b/requirements-nvidia-cu130.txt @@ -14,7 +14,7 @@ # # For manual installation, run these commands in order: # pip install -r requirements-nvidia-cu130.txt -# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 +# pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 # ============================================================================ # --- PyTorch CUDA 13.0 Package Index --- diff --git a/requirements-nvidia.txt b/requirements-nvidia.txt index e8d64a3..73e26bc 100644 --- a/requirements-nvidia.txt +++ b/requirements-nvidia.txt @@ -20,7 +20,7 @@ torchaudio==2.5.1+cu121 # # For manual installation, run these commands in order: # pip install -r requirements-nvidia.txt -# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 +# pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 # Chatterbox Dependencies (explicit, since --no-deps skips them) # NOTE: s3tokenizer and onnx are installed with --no-deps alongside chatterbox diff --git a/requirements-rdna4-init.txt b/requirements-rdna4-init.txt index b109996..8567326 100644 --- a/requirements-rdna4-init.txt +++ b/requirements-rdna4-init.txt @@ -10,7 +10,7 @@ # Manual install: # pip install -r requirements-rdna4-init.txt # pip install -r requirements-rdna4.txt -# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 +# pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.3/triton-3.5.1%2Brocm7.2.3.gita272dfa8-cp310-cp310-linux_x86_64.whl https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.3/torch-2.9.1%2Brocm7.2.3.lw.gitebc02d69-cp310-cp310-linux_x86_64.whl diff --git a/requirements-rdna4.txt b/requirements-rdna4.txt index 7c55ef0..a14ae7f 100644 --- a/requirements-rdna4.txt +++ b/requirements-rdna4.txt @@ -10,7 +10,7 @@ # Manual installation order: # pip install -r requirements-rdna4-init.txt # pip install -r requirements-rdna4.txt -# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 +# pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 # # The --no-deps flag on chatterbox-tts is critical: without it, pip will # replace the ROCm torch wheels with CPU-only versions from PyPI. diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt index a9e3d03..afaac1d 100644 --- a/requirements-rocm-init.txt +++ b/requirements-rocm-init.txt @@ -8,7 +8,7 @@ # For manual installation: # pip install -r requirements-rocm-init.txt # pip install -r requirements-rocm.txt -# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master +# pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 -i https://download.pytorch.org/whl/rocm6.1 torch==2.5.1 diff --git a/requirements-rocm.txt b/requirements-rocm.txt index e586583..d117976 100644 --- a/requirements-rocm.txt +++ b/requirements-rocm.txt @@ -9,7 +9,7 @@ # For manual installation, run these commands in order: # pip install -r requirements-rocm-init.txt # pip install -r requirements-rocm.txt -# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 +# pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 # # The --no-deps flag on chatterbox-tts is critical: without it, pip will # replace the ROCm torch wheels with CPU-only versions from PyPI. diff --git a/requirements-strixhalo.txt b/requirements-strixhalo.txt index 0aaa939..23f6853 100644 --- a/requirements-strixhalo.txt +++ b/requirements-strixhalo.txt @@ -1,7 +1,7 @@ # requirements.txt -# Chatterbox TTS engine - Install from chatterbox-v2 fork -chatterbox-tts @ git+https://github.com/devnen/chatterbox-v2.git@master +# Chatterbox TTS engine +chatterbox-tts @ git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 # Core Web Framework fastapi diff --git a/requirements.txt b/requirements.txt index 23576b7..969c2b4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -21,7 +21,7 @@ torchaudio==2.5.1 # # For manual installation, run these commands in order: # pip install -r requirements.txt -# pip install --no-deps git+https://github.com/devnen/chatterbox-v2.git@master s3tokenizer==0.3.0 onnx==1.16.0 +# pip install --no-deps git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2 s3tokenizer==0.3.0 onnx==1.16.0 # --- Chatterbox Dependencies (explicit, since --no-deps skips them) --- # NOTE: s3tokenizer and onnx are installed with --no-deps alongside chatterbox diff --git a/start.py b/start.py index 728a154..a107326 100644 --- a/start.py +++ b/start.py @@ -109,7 +109,7 @@ } # Chatterbox fork URL (used for CUDA 12.8 installation) -CHATTERBOX_REPO = "git+https://github.com/devnen/chatterbox-v2.git@master" +CHATTERBOX_REPO = "git+https://github.com/resemble-ai/chatterbox.git@5de7a54aa4e5e2baadb0182dde554908b48b85c2" # Timeout settings (seconds) # First run downloads large model files (~2GB). Subsequent starts are much faster.