fix typo

pytorch · Apr 6, 2024 · 304c52f · 304c52f
1 parent 7c5c7bf
commit 304c52f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/export_aoti.py b/export_aoti.py
@@ -46,7 +46,7 @@ def export_model(
     # with torch.device(device):
     #     model.setup_caches(max_batch_size=1, max_seq_length=max_seq_length)
 
-    device=
+    device="cpu"
     input = (
         torch.tensor([[1, 9038, 2501,  263,  931]], dtype=torch.int, device=device),
         torch.tensor([0, 1, 2, 3, 4], dtype=torch.int, device=device),