seemanthraju
Claude Opus 4.5
commited on
Commit
·
7189a0b
1
Parent(s):
ffa489d
Fix HuggingFace repo name and model file paths
Browse files- Update DEFAULT_HF_REPO from Seemanth/chiluka-tts to Seemanth/chiluka
- Fix model registry paths to include chiluka/ prefix matching HF repo structure
- Update all references across README, hubconf, and inference module
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
- README.md +2 -2
- README_HF.md +1 -1
- chiluka/hub.py +13 -13
- chiluka/inference.py +1 -1
- hubconf.py +1 -1
README.md
CHANGED
|
@@ -177,7 +177,7 @@ tts = Chiluka.from_pretrained(model="hindi_english") # Hindi-English (ex
|
|
| 177 |
# With options
|
| 178 |
tts = Chiluka.from_pretrained(
|
| 179 |
model="hindi_english", # Model variant
|
| 180 |
-
repo_id="Seemanth/chiluka
|
| 181 |
device="cuda", # or "cpu"
|
| 182 |
force_download=False, # Re-download even if cached
|
| 183 |
token="hf_xxx" # For private repos
|
|
@@ -252,7 +252,7 @@ list_models()
|
|
| 252 |
|
| 253 |
# Clear cache
|
| 254 |
clear_cache() # Clear all
|
| 255 |
-
clear_cache("Seemanth/chiluka
|
| 256 |
|
| 257 |
# Push your own model to HuggingFace
|
| 258 |
push_to_hub(
|
|
|
|
| 177 |
# With options
|
| 178 |
tts = Chiluka.from_pretrained(
|
| 179 |
model="hindi_english", # Model variant
|
| 180 |
+
repo_id="Seemanth/chiluka", # HuggingFace repo
|
| 181 |
device="cuda", # or "cpu"
|
| 182 |
force_download=False, # Re-download even if cached
|
| 183 |
token="hf_xxx" # For private repos
|
|
|
|
| 252 |
|
| 253 |
# Clear cache
|
| 254 |
clear_cache() # Clear all
|
| 255 |
+
clear_cache("Seemanth/chiluka") # Clear specific repo
|
| 256 |
|
| 257 |
# Push your own model to HuggingFace
|
| 258 |
push_to_hub(
|
README_HF.md
CHANGED
|
@@ -63,7 +63,7 @@ wav = tts.synthesize("Hello!", "reference.wav", language="en")
|
|
| 63 |
```python
|
| 64 |
from chiluka import Chiluka
|
| 65 |
|
| 66 |
-
tts = Chiluka.from_pretrained("Seemanth/chiluka
|
| 67 |
```
|
| 68 |
|
| 69 |
## Parameters
|
|
|
|
| 63 |
```python
|
| 64 |
from chiluka import Chiluka
|
| 65 |
|
| 66 |
+
tts = Chiluka.from_pretrained("Seemanth/chiluka")
|
| 67 |
```
|
| 68 |
|
| 69 |
## Parameters
|
chiluka/hub.py
CHANGED
|
@@ -14,7 +14,7 @@ from pathlib import Path
|
|
| 14 |
from typing import Optional, Union
|
| 15 |
|
| 16 |
# Default HuggingFace Hub repository
|
| 17 |
-
DEFAULT_HF_REPO = "Seemanth/chiluka
|
| 18 |
|
| 19 |
# Cache directory for downloaded models
|
| 20 |
CACHE_DIR = Path.home() / ".cache" / "chiluka"
|
|
@@ -26,14 +26,14 @@ CACHE_DIR = Path.home() / ".cache" / "chiluka"
|
|
| 26 |
# relative to the repo root.
|
| 27 |
MODEL_REGISTRY = {
|
| 28 |
"telugu": {
|
| 29 |
-
"config": "configs/config_ft.yml",
|
| 30 |
-
"checkpoint": "checkpoints/epoch_2nd_00017.pth",
|
| 31 |
"languages": ["te", "en"],
|
| 32 |
"description": "Telugu + English single-speaker TTS",
|
| 33 |
},
|
| 34 |
"hindi_english": {
|
| 35 |
-
"config": "configs/config_hindi_english.yml",
|
| 36 |
-
"checkpoint": "checkpoints/epoch_2nd_00029.pth",
|
| 37 |
"languages": ["hi", "en"],
|
| 38 |
"description": "Hindi + English multi-speaker TTS (5 speakers)",
|
| 39 |
},
|
|
@@ -43,11 +43,11 @@ DEFAULT_MODEL = "hindi_english"
|
|
| 43 |
|
| 44 |
# Shared pretrained sub-models (same across all variants)
|
| 45 |
PRETRAINED_FILES = {
|
| 46 |
-
"asr_config": "pretrained/ASR/config.yml",
|
| 47 |
-
"asr_model": "pretrained/ASR/epoch_00080.pth",
|
| 48 |
-
"f0_model": "pretrained/JDC/bst.t7",
|
| 49 |
-
"plbert_config": "pretrained/PLBERT/config.yml",
|
| 50 |
-
"plbert_model": "pretrained/PLBERT/step_1000000.t7",
|
| 51 |
}
|
| 52 |
|
| 53 |
|
|
@@ -110,7 +110,7 @@ def download_from_hf(
|
|
| 110 |
Download model files from HuggingFace Hub.
|
| 111 |
|
| 112 |
Args:
|
| 113 |
-
repo_id: HuggingFace Hub repository ID (e.g., 'Seemanth/chiluka
|
| 114 |
revision: Git revision to download (branch, tag, or commit hash)
|
| 115 |
force_download: If True, re-download even if cached
|
| 116 |
token: HuggingFace API token for private repos
|
|
@@ -173,7 +173,7 @@ def get_model_paths(
|
|
| 173 |
return {
|
| 174 |
"config_path": str(model_dir / model_info["config"]),
|
| 175 |
"checkpoint_path": str(model_dir / model_info["checkpoint"]),
|
| 176 |
-
"pretrained_dir": str(model_dir / "pretrained"),
|
| 177 |
}
|
| 178 |
|
| 179 |
|
|
@@ -218,7 +218,7 @@ def push_to_hub(
|
|
| 218 |
Example:
|
| 219 |
>>> push_to_hub(
|
| 220 |
... local_dir="./chiluka",
|
| 221 |
-
... repo_id="Seemanth/chiluka
|
| 222 |
... private=False
|
| 223 |
... )
|
| 224 |
"""
|
|
|
|
| 14 |
from typing import Optional, Union
|
| 15 |
|
| 16 |
# Default HuggingFace Hub repository
|
| 17 |
+
DEFAULT_HF_REPO = "Seemanth/chiluka"
|
| 18 |
|
| 19 |
# Cache directory for downloaded models
|
| 20 |
CACHE_DIR = Path.home() / ".cache" / "chiluka"
|
|
|
|
| 26 |
# relative to the repo root.
|
| 27 |
MODEL_REGISTRY = {
|
| 28 |
"telugu": {
|
| 29 |
+
"config": "chiluka/configs/config_ft.yml",
|
| 30 |
+
"checkpoint": "chiluka/checkpoints/epoch_2nd_00017.pth",
|
| 31 |
"languages": ["te", "en"],
|
| 32 |
"description": "Telugu + English single-speaker TTS",
|
| 33 |
},
|
| 34 |
"hindi_english": {
|
| 35 |
+
"config": "chiluka/configs/config_hindi_english.yml",
|
| 36 |
+
"checkpoint": "chiluka/checkpoints/epoch_2nd_00029.pth",
|
| 37 |
"languages": ["hi", "en"],
|
| 38 |
"description": "Hindi + English multi-speaker TTS (5 speakers)",
|
| 39 |
},
|
|
|
|
| 43 |
|
| 44 |
# Shared pretrained sub-models (same across all variants)
|
| 45 |
PRETRAINED_FILES = {
|
| 46 |
+
"asr_config": "chiluka/pretrained/ASR/config.yml",
|
| 47 |
+
"asr_model": "chiluka/pretrained/ASR/epoch_00080.pth",
|
| 48 |
+
"f0_model": "chiluka/pretrained/JDC/bst.t7",
|
| 49 |
+
"plbert_config": "chiluka/pretrained/PLBERT/config.yml",
|
| 50 |
+
"plbert_model": "chiluka/pretrained/PLBERT/step_1000000.t7",
|
| 51 |
}
|
| 52 |
|
| 53 |
|
|
|
|
| 110 |
Download model files from HuggingFace Hub.
|
| 111 |
|
| 112 |
Args:
|
| 113 |
+
repo_id: HuggingFace Hub repository ID (e.g., 'Seemanth/chiluka')
|
| 114 |
revision: Git revision to download (branch, tag, or commit hash)
|
| 115 |
force_download: If True, re-download even if cached
|
| 116 |
token: HuggingFace API token for private repos
|
|
|
|
| 173 |
return {
|
| 174 |
"config_path": str(model_dir / model_info["config"]),
|
| 175 |
"checkpoint_path": str(model_dir / model_info["checkpoint"]),
|
| 176 |
+
"pretrained_dir": str(model_dir / "chiluka" / "pretrained"),
|
| 177 |
}
|
| 178 |
|
| 179 |
|
|
|
|
| 218 |
Example:
|
| 219 |
>>> push_to_hub(
|
| 220 |
... local_dir="./chiluka",
|
| 221 |
+
... repo_id="Seemanth/chiluka",
|
| 222 |
... private=False
|
| 223 |
... )
|
| 224 |
"""
|
chiluka/inference.py
CHANGED
|
@@ -172,7 +172,7 @@ class Chiluka:
|
|
| 172 |
model: Model variant to load. Options:
|
| 173 |
- 'hindi_english' (default) - Hindi + English multi-speaker TTS
|
| 174 |
- 'telugu' - Telugu + English single-speaker TTS
|
| 175 |
-
repo_id: HuggingFace Hub repository ID (e.g., 'Seemanth/chiluka
|
| 176 |
If None, uses the default repository.
|
| 177 |
device: Device to use ('cuda' or 'cpu'). Auto-detects if None.
|
| 178 |
force_download: If True, re-download even if cached.
|
|
|
|
| 172 |
model: Model variant to load. Options:
|
| 173 |
- 'hindi_english' (default) - Hindi + English multi-speaker TTS
|
| 174 |
- 'telugu' - Telugu + English single-speaker TTS
|
| 175 |
+
repo_id: HuggingFace Hub repository ID (e.g., 'Seemanth/chiluka').
|
| 176 |
If None, uses the default repository.
|
| 177 |
device: Device to use ('cuda' or 'cpu'). Auto-detects if None.
|
| 178 |
force_download: If True, re-download even if cached.
|
hubconf.py
CHANGED
|
@@ -98,7 +98,7 @@ def chiluka_hindi_english(pretrained: bool = True, device: str = None, **kwargs)
|
|
| 98 |
return chiluka(pretrained=pretrained, device=device, **kwargs)
|
| 99 |
|
| 100 |
|
| 101 |
-
def chiluka_from_hf(repo_id: str = "Seemanth/chiluka
|
| 102 |
"""
|
| 103 |
Load Chiluka TTS from a specific HuggingFace Hub repository.
|
| 104 |
|
|
|
|
| 98 |
return chiluka(pretrained=pretrained, device=device, **kwargs)
|
| 99 |
|
| 100 |
|
| 101 |
+
def chiluka_from_hf(repo_id: str = "Seemanth/chiluka", model: str = "hindi_english", device: str = None, **kwargs):
|
| 102 |
"""
|
| 103 |
Load Chiluka TTS from a specific HuggingFace Hub repository.
|
| 104 |
|