Tomtom84 commited on
Commit
f2002a0
·
verified ·
1 Parent(s): c9fb95e

Update engines/orpheus_engine.py

Browse files
Files changed (1) hide show
  1. engines/orpheus_engine.py +2 -1
engines/orpheus_engine.py CHANGED
@@ -14,6 +14,7 @@ DEFAULT_API_URL = "http://127.0.0.1:1234"
14
  DEFAULT_HEADERS = {"Content-Type": "application/json"}
15
  DEFAULT_MODEL = "SebastianBodza/Kartoffel_Orpheus-3B_german_synthetic-v0.1"
16
  DEFAULT_VOICE = "Martin"
 
17
  SAMPLE_RATE = 24000 # Specific sample rate for Orpheus
18
 
19
  # Special token definitions for prompt formatting and token decoding
@@ -185,7 +186,7 @@ class OrpheusEngine(BaseEngine):
185
 
186
  payload = {
187
  "model": self.model,
188
- "messages": [{"role": "user", "content": formatted_prompt}],
189
  "max_tokens": self.max_tokens,
190
  "temperature": self.temperature,
191
  "top_p": self.top_p,
 
14
  DEFAULT_HEADERS = {"Content-Type": "application/json"}
15
  DEFAULT_MODEL = "SebastianBodza/Kartoffel_Orpheus-3B_german_synthetic-v0.1"
16
  DEFAULT_VOICE = "Martin"
17
+ STOP_SEQUENCE = "<custom_token_2>"
18
  SAMPLE_RATE = 24000 # Specific sample rate for Orpheus
19
 
20
  # Special token definitions for prompt formatting and token decoding
 
186
 
187
  payload = {
188
  "model": self.model,
189
+ "messages": [{"role": "user", "content": f"<|audio|>{voice}: {text}<|eot_id|>"}],
190
  "max_tokens": self.max_tokens,
191
  "temperature": self.temperature,
192
  "top_p": self.top_p,