hfhfh

orpheus-tts/engine_class.py

@@ -150,7 +150,7 @@ class OrpheusModel:
                     return prompt_string
 
  
-    def generate_tokens_sync(self, prompt, voice=None, request_id="req-001", temperature=0.6, top_p=0.95, max_tokens=4000, stop_token_ids = [128258], repetition_penalty=1.1):
+    def generate_tokens_sync(self, prompt, voice=None, request_id="req-001", temperature=0.6, top_p=0.95, max_tokens=1000, stop_token_ids=[128258], repetition_penalty=1.1):
         print(f"DEBUG: Original prompt: {prompt}")
         print(f"DEBUG: Voice: {voice}")
         print(f"DEBUG: Model name: {self.model_name}")
@@ -161,7 +161,7 @@ class OrpheusModel:
         temperature=temperature,
         top_p=top_p,
         max_tokens=max_tokens,  # Adjust max_tokens as needed.
-        stop_token_ids = stop_token_ids,
+        stop_token_ids=stop_token_ids,
         repetition_penalty=repetition_penalty,
         )