Update app.py
Browse files
app.py
CHANGED
@@ -77,11 +77,11 @@ def build_prompt(text: str, voice: str):
|
|
77 |
|
78 |
def decode_block(block7: list[int]) -> bytes:
|
79 |
l1,l2,l3=[],[],[]
|
80 |
-
l1.append(block7[0]
|
81 |
-
l2.append(block7[1]
|
82 |
-
l3 += [block7[2]
|
83 |
-
l2.append(block7[4]
|
84 |
-
l3 += [block7[5]
|
85 |
|
86 |
with torch.no_grad():
|
87 |
codes = [torch.tensor(x, device=device).unsqueeze(0)
|
|
|
77 |
|
78 |
def decode_block(block7: list[int]) -> bytes:
|
79 |
l1,l2,l3=[],[],[]
|
80 |
+
l1.append(block7[0] - 0 * 4096) # Subtract position 0 offset
|
81 |
+
l2.append(block7[1] - 1 * 4096) # Subtract position 1 offset
|
82 |
+
l3 += [block7[2] - 2 * 4096, block7[3] - 3 * 4096] # Subtract position offsets
|
83 |
+
l2.append(block7[4] - 4 * 4096) # Subtract position 4 offset
|
84 |
+
l3 += [block7[5] - 5 * 4096, block7[6] - 6 * 4096] # Subtract position offsets
|
85 |
|
86 |
with torch.no_grad():
|
87 |
codes = [torch.tensor(x, device=device).unsqueeze(0)
|