From 95d45e590f39bdf14031743e98770302d01ed570 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Mon, 16 Dec 2024 12:52:04 +0100 Subject: [PATCH] test again --- optimum_benchmark/backends/py_txi/backend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum_benchmark/backends/py_txi/backend.py b/optimum_benchmark/backends/py_txi/backend.py index 96a80b6c..a3341bcf 100644 --- a/optimum_benchmark/backends/py_txi/backend.py +++ b/optimum_benchmark/backends/py_txi/backend.py @@ -67,8 +67,8 @@ def create_no_weights_model(self) -> None: self.generation_config.save_pretrained(save_directory=self.no_weights_model) def load_model_with_no_weights(self) -> None: - self.config.volumes = {self.tmpdir.name: {"bind": "/var/no_weights_folder", "mode": "rw"}} - original_model, self.config.model = self.config.model, "/var/no_weights_folder/no_weights_model" + self.config.volumes = {self.tmpdir.name: {"bind": "/data", "mode": "rw"}} + original_model, self.config.model = self.config.model, "/data/no_weights_model/" self.load_model_from_pretrained() self.config.model = original_model