Tomtom84 commited on
Commit
af8d415
·
verified ·
1 Parent(s): b3159ca

Update engines/orpheus_decoder.py

Browse files
Files changed (1) hide show
  1. engines/orpheus_decoder.py +1 -1
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 = []