Update engines/orpheus_engine.py
Browse files
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.
|
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
|