rahul7star commited on
Commit
d0da5ef
·
verified ·
1 Parent(s): 9a23465

Update web_ui.py

Browse files
Files changed (1) hide show
  1. web_ui.py +6 -5
web_ui.py CHANGED
@@ -29,12 +29,13 @@ def initialize_model():
29
  model_dir = "YatharthS/MiraTTS"
30
 
31
  # Initialize with HF Spaces compatible settings
32
- MODEL = MiraTTS( model_dir=model_dir,
33
- tp=1,
34
- device_type="cpu", # use CPU only
35
- enable_prefix_caching=False,
36
- cache_max_entry_count=0.0
37
  )
 
38
 
39
 
40
  logging.info("Model initialized successfully")
 
29
  model_dir = "YatharthS/MiraTTS"
30
 
31
  # Initialize with HF Spaces compatible settings
32
+ MODEL = MiraTTS(
33
+ model_dir=model_dir,
34
+ tp=1, # Single GPU
35
+ enable_prefix_caching=False, # Disable for stability
36
+ cache_max_entry_count=0.1 # Reduced cache
37
  )
38
+
39
 
40
 
41
  logging.info("Model initialized successfully")