From df394f8b40518116f26cd2a2e21c6373ad1b6ded Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 02:02:14 +0000 Subject: [PATCH] fix: missing closing quote on string literal --- single_shot_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/single_shot_inference.py b/single_shot_inference.py index 1f0c528d..bb8c4c5e 100644 --- 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