diff --git a/optimum_benchmark/backends/py_txi/backend.py b/optimum_benchmark/backends/py_txi/backend.py index e7e298f4..3acd2e42 100644 --- a/optimum_benchmark/backends/py_txi/backend.py +++ b/optimum_benchmark/backends/py_txi/backend.py @@ -39,7 +39,7 @@ def load(self) -> None: try: self.tmpdir.cleanup() except Exception: - shutil.rmtree(self.tmpdir.name) + shutil.rmtree(self.tmpdir.name, ignore_errors=True) def download_pretrained_model(self) -> None: model_snapshot_folder = snapshot_download(self.config.model, **self.config.model_kwargs)