Skip to content

Commit 062b485

Browse files
authored
Merge branch 'develop' into fx/metatypes
2 parents ecae4d8 + 7ed1617 commit 062b485

File tree

1 file changed

+1
-1
lines changed
  • tests/post_training/pipelines

1 file changed

+1
-1
lines changed

tests/post_training/pipelines/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -447,7 +447,7 @@ def save_compressed_model(self) -> None:
447447
self.path_compressed_ir = self.output_model_dir / "model.xml"
448448
ov.serialize(ov_model, self.path_compressed_ir)
449449

450-
if BackendType.CUDA_FX_TORCH:
450+
if self.backend == BackendType.CUDA_FX_TORCH:
451451
self.model = self.model.cuda()
452452
self.dummy_tensor = self.dummy_tensor.cuda()
453453

0 commit comments

Comments
 (0)