Update engines/orpheus_decoder.py
Browse files
engines/orpheus_decoder.py
CHANGED
|
@@ -11,7 +11,7 @@ model = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").eval()
|
|
| 11 |
# Check if CUDA is available and set device accordingly
|
| 12 |
snac_device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
|
| 13 |
model = model.to(snac_device)
|
| 14 |
-
|
| 15 |
|
| 16 |
def convert_to_audio(multiframe, count):
|
| 17 |
frames = []
|
|
|
|
| 11 |
# Check if CUDA is available and set device accordingly
|
| 12 |
snac_device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
|
| 13 |
model = model.to(snac_device)
|
| 14 |
+
snac_device = "cuda"
|
| 15 |
|
| 16 |
def convert_to_audio(multiframe, count):
|
| 17 |
frames = []
|