Skip to content

Commit 5872433

Browse files
committed
manually fix rebase issues
1 parent ada3e08 commit 5872433

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

recipes_source/torch_export_aoti_python.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@
176176
model_path = os.path.join(os.getcwd(), "resnet18.pt2")
177177

178178
compiled_model = torch._inductor.aoti_load_package(model_path)
179-
example_inputs = torch.randn(2, 3, 224, 224, device=device)
179+
example_inputs = (torch.randn(2, 3, 224, 224, device=device),)
180180

181181
with torch.inference_mode():
182182
output = compiled_model(example_inputs)

0 commit comments

Comments
 (0)