Fix: Add missing CACHE_DIR to data.manager for training cache support
Browse files- data/manager.py +3 -1
data/manager.py
CHANGED
|
@@ -8,12 +8,14 @@ DATA_DIR = os.path.abspath(DATA_DIR)
|
|
| 8 |
os.makedirs(DATA_DIR, exist_ok=True)
|
| 9 |
|
| 10 |
# Export BASE_DIR for use in other modules
|
| 11 |
-
__all__ = ['DATA_DIR', 'AUDIO_DIR', 'ENTITIES_PATH', 'MODEL_OUTPUT_DIR', 'BASE_DIR']
|
| 12 |
|
| 13 |
AUDIO_DIR = os.path.join(DATA_DIR, "audio_files")
|
| 14 |
os.makedirs(AUDIO_DIR, exist_ok=True)
|
| 15 |
ENTITIES_PATH = os.path.join(DATA_DIR, "caribbean_entities.json")
|
| 16 |
MODEL_OUTPUT_DIR = os.path.join(DATA_DIR, "owsm_caribbean_finetuned")
|
|
|
|
|
|
|
| 17 |
|
| 18 |
# Debug paths
|
| 19 |
print(f"DEBUG: DATA_DIR = {DATA_DIR}")
|
|
|
|
| 8 |
os.makedirs(DATA_DIR, exist_ok=True)
|
| 9 |
|
| 10 |
# Export BASE_DIR for use in other modules
|
| 11 |
+
__all__ = ['DATA_DIR', 'AUDIO_DIR', 'ENTITIES_PATH', 'MODEL_OUTPUT_DIR', 'BASE_DIR', 'CACHE_DIR']
|
| 12 |
|
| 13 |
AUDIO_DIR = os.path.join(DATA_DIR, "audio_files")
|
| 14 |
os.makedirs(AUDIO_DIR, exist_ok=True)
|
| 15 |
ENTITIES_PATH = os.path.join(DATA_DIR, "caribbean_entities.json")
|
| 16 |
MODEL_OUTPUT_DIR = os.path.join(DATA_DIR, "owsm_caribbean_finetuned")
|
| 17 |
+
CACHE_DIR = os.path.join(DATA_DIR, "processed_datasets_cache")
|
| 18 |
+
os.makedirs(CACHE_DIR, exist_ok=True)
|
| 19 |
|
| 20 |
# Debug paths
|
| 21 |
print(f"DEBUG: DATA_DIR = {DATA_DIR}")
|