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 = []
|