init

anzr299 · anzr299 · commit 8b9e37c96b2a · 2025-04-24T17:31:05.000+04:00
diff --git a/tests/post_training/pipelines/base.py b/tests/post_training/pipelines/base.py
@@ -506,7 +506,7 @@ def save_compressed_model(self) -> None:
             mod = torch.compile(
                 exported_model.module(),
                 backend="openvino",
-                options={"model_caching": True, "cache_dir": str(self.output_model_dir)},
+                options={"model_caching": True, "cache_dir": str(self.output_model_dir), "aot_autograd": True},
             )
             mod(self.dummy_tensor)
 

Original file line number	Diff line number	Diff line change
`@@ -506,7 +506,7 @@ def save_compressed_model(self) -> None:`
`506`	`506`	`mod = torch.compile(`
`507`	`507`	`exported_model.module(),`
`508`	`508`	`backend="openvino",`
`509`		`- options={"model_caching": True, "cache_dir": str(self.output_model_dir)},`
	`509`	`+ options={"model_caching": True, "cache_dir": str(self.output_model_dir), "aot_autograd": True},`
`510`	`510`	`)`
`511`	`511`	`mod(self.dummy_tensor)`
`512`	`512`