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>

Files changed (5) hide show
  1. README.md +2 -2
  2. README_HF.md +1 -1
  3. chiluka/hub.py +13 -13
  4. chiluka/inference.py +1 -1
  5. 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-tts", # HuggingFace repo
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-tts") # Clear specific repo
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-tts")
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-tts"
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-tts')
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-tts",
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-tts').
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-tts", model: str = "hindi_english", device: str = None, **kwargs):
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