Tomtom84 commited on
Commit
bd409ef
·
1 Parent(s): e668e7f
orpheus-tts/__pycache__/kartoffel_decoder.cpython-310.pyc CHANGED
Binary files a/orpheus-tts/__pycache__/kartoffel_decoder.cpython-310.pyc and b/orpheus-tts/__pycache__/kartoffel_decoder.cpython-310.pyc differ
 
orpheus-tts/kartoffel_decoder.py CHANGED
@@ -66,9 +66,9 @@ def redistribute_codes_kartoffel(code_list):
66
  return torch.tensor([[]], device=snac_device, dtype=torch.float32)
67
 
68
  codes = [
69
- torch.tensor(layer_1, device=snac_device).unsqueeze(0),
70
- torch.tensor(layer_2, device=snac_device).unsqueeze(0),
71
- torch.tensor(layer_3, device=snac_device).unsqueeze(0),
72
  ]
73
 
74
  with torch.no_grad():
 
66
  return torch.tensor([[]], device=snac_device, dtype=torch.float32)
67
 
68
  codes = [
69
+ torch.tensor(layer_1, device=snac_device, dtype=torch.int32).unsqueeze(0),
70
+ torch.tensor(layer_2, device=snac_device, dtype=torch.int32).unsqueeze(0),
71
+ torch.tensor(layer_3, device=snac_device, dtype=torch.int32).unsqueeze(0),
72
  ]
73
 
74
  with torch.no_grad():