Tomtom84 commited on
Commit
549abbb
·
verified ·
1 Parent(s): 0d60bb9

Update engines/orpheus_engine.py

Browse files
Files changed (1) hide show
  1. engines/orpheus_engine.py +1 -1
engines/orpheus_engine.py CHANGED
@@ -100,7 +100,7 @@ def _fade_in_out(audio: np.ndarray, fade_ms: int = 50) -> np.ndarray:
100
  try:
101
  from snac import SNAC
102
  _snac_model: Optional[SNAC] = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").eval()
103
- _snac_model = _snac_model.to("cuda" if _snac_model and _snac_model.is_cuda_available() else "cpu")
104
  except Exception as exc: # pragma: no cover
105
  logging.warning("SNAC model could not be loaded – %s", exc)
106
  _snac_model = None
 
100
  try:
101
  from snac import SNAC
102
  _snac_model: Optional[SNAC] = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").eval()
103
+ _snac_model = _snac_model.to("cuda" if _snac_model and _snac_model.torch.cuda.is_available() else "cpu")
104
  except Exception as exc: # pragma: no cover
105
  logging.warning("SNAC model could not be loaded – %s", exc)
106
  _snac_model = None