We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents ecae4d8 + 7ed1617 commit 062b485Copy full SHA for 062b485
tests/post_training/pipelines/base.py
@@ -447,7 +447,7 @@ def save_compressed_model(self) -> None:
447
self.path_compressed_ir = self.output_model_dir / "model.xml"
448
ov.serialize(ov_model, self.path_compressed_ir)
449
450
- if BackendType.CUDA_FX_TORCH:
+ if self.backend == BackendType.CUDA_FX_TORCH:
451
self.model = self.model.cuda()
452
self.dummy_tensor = self.dummy_tensor.cuda()
453
0 commit comments