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)
|