Spaces:
Running
on
Zero
Running
on
Zero
Two calls in a row
Browse files
app.py
CHANGED
@@ -65,6 +65,8 @@ def run_model():
|
|
65 |
if compiled_model is None:
|
66 |
compiled_model = torch._inductor.aoti_load_package(package_path)
|
67 |
compiled_model.load_constants(weights, check_full_update=True, user_managed=True)
|
|
|
|
|
68 |
with torch.inference_mode():
|
69 |
return str(compiled_model(example_inputs))
|
70 |
|
|
|
65 |
if compiled_model is None:
|
66 |
compiled_model = torch._inductor.aoti_load_package(package_path)
|
67 |
compiled_model.load_constants(weights, check_full_update=True, user_managed=True)
|
68 |
+
with torch.inference_mode():
|
69 |
+
compiled_model(example_inputs)
|
70 |
with torch.inference_mode():
|
71 |
return str(compiled_model(example_inputs))
|
72 |
|