Update mira/model.py

mira/model.py

@@ -11,6 +11,7 @@ class MiraTTS_GPU:
     def __init__(self, model_dir="rahul7star/mir-TTS", tp=1, enable_prefix_caching=True, cache_max_entry_count=0.2):
         
         backend_config = TurbomindEngineConfig(cache_max_entry_count=cache_max_entry_count, tp=tp, dtype='bfloat16', enable_prefix_caching=enable_prefix_caching)
+        
         self.pipe = pipeline(model_dir, backend_config=backend_config)
         self.gen_config = GenerationConfig(top_p=0.95,
                               top_k=50,
@@ -86,11 +87,13 @@ class MiraTTS:
 
     def __init__(self, model_dir="rahul7star/mir-TTS", tp=1, enable_prefix_caching=True, cache_max_entry_count=0.2):
         # Use float32 for CPU
+       
+
+
         backend_config = TurbomindEngineConfig(
-            cache_max_entry_count=cache_max_entry_count,
-            tp=tp,
+            
             dtype="float32",  # CPU compatible
-            enable_prefix_caching=enable_prefix_caching
+            cache_max_entry_count=0.0
         )
         self.pipe = pipeline(model_dir, backend_config=backend_config, device="cpu")