fix: missing closing quote on string literal

2026-06-01 02:02:14 +00:00
parent cfd2468c61
commit df394f8b40
1 changed files with 1 additions and 1 deletions
--- a/single_shot_inference.py
+++ b/single_shot_inference.py
@@ -717,7 +717,7 @@ def main():
        t1 = time.time()
        tid_int64 = torch.tensor([all_tokens[-1]], dtype=torch.long, device='cuda:0')
        tid = tid_int64.to(torch.int32)  # hash router needs int32
-        dec_pos = torch.tensor([len(all_tokens)-1], dtype=torch.long, device='cuda:0)
+        dec_pos = torch.tensor([len(all_tokens)-1], dtype=torch.long, device='cuda:0')
        X = mHCLayer.init_state(embed(tid_int64))
        for li in range(n_layers):
            gpu = li % NUM_GPUS