Tomtom84 commited on
Commit
0d60bb9
·
verified ·
1 Parent(s): 8ee26de

Update engines/orpheus_engine.py

Browse files
Files changed (1) hide show
  1. engines/orpheus_engine.py +17 -17
engines/orpheus_engine.py CHANGED
@@ -64,22 +64,7 @@ _STEADY_GROUPS = 30 # 30×7 = 210 codes ≈ 900 ms
64
  SNAC_MODEL = os.getenv("SNAC_MODEL", "hubertsiuzdak/snac_24khz")
65
 
66
 
67
- def _load_snac(self, model_name: str = SNAC_MODEL):
68
- """
69
- Lädt den SNAC-Decoder auf CPU/GPU.
70
- Fällt bei jedem Fehler sauber auf CPU zurück.
71
- """
72
- device = "cuda" if torch.cuda.is_available() else "cpu"
73
- try:
74
- snac = SNAC.from_pretrained(model_name).to(device)
75
- if device == "cuda": # half() nur auf GPU – ältere SNAC-Versionen haben keine .half()
76
- snac = snac.half()
77
- snac.eval()
78
- logging.info(f"SNAC {snac_version} loaded on {device}")
79
- return snac
80
- except Exception as e:
81
- logging.exception("SNAC load failed – running with silent fallback")
82
- return None
83
 
84
  ###############################################################################
85
  # Helper functions #
@@ -185,7 +170,22 @@ class OrpheusEngine(BaseEngine):
185
  OrpheusVoice("Sophia", "f"), OrpheusVoice("Lina", "f"),
186
  OrpheusVoice("Lea", "f"),
187
  ]
188
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
189
  # ---------------------------------------------------------------------
190
  def __init__(
191
  self,
 
64
  SNAC_MODEL = os.getenv("SNAC_MODEL", "hubertsiuzdak/snac_24khz")
65
 
66
 
67
+
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
68
 
69
  ###############################################################################
70
  # Helper functions #
 
170
  OrpheusVoice("Sophia", "f"), OrpheusVoice("Lina", "f"),
171
  OrpheusVoice("Lea", "f"),
172
  ]
173
+ def _load_snac(self, model_name: str = SNAC_MODEL):
174
+ """
175
+ Lädt den SNAC-Decoder auf CPU/GPU.
176
+ Fällt bei jedem Fehler sauber auf CPU zurück.
177
+ """
178
+ device = "cuda" if torch.cuda.is_available() else "cpu"
179
+ try:
180
+ snac = SNAC.from_pretrained(model_name).to(device)
181
+ if device == "cuda": # half() nur auf GPU – ältere SNAC-Versionen haben keine .half()
182
+ snac = snac.half()
183
+ snac.eval()
184
+ logging.info(f"SNAC {snac_version} loaded on {device}")
185
+ return snac
186
+ except Exception as e:
187
+ logging.exception("SNAC load failed – running with silent fallback")
188
+ return None
189
  # ---------------------------------------------------------------------
190
  def __init__(
191
  self,