shaun3141 commited on
Commit
b903db2
·
1 Parent(s): c0cd25b

Fix: Add missing CACHE_DIR to data.manager for training cache support

Browse files
Files changed (1) hide show
  1. 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}")