test

huggingface · Dec 16, 2024 · 9d5cf6c · 9d5cf6c
1 parent 942f0e0
commit 9d5cf6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum_benchmark/backends/py_txi/backend.py b/optimum_benchmark/backends/py_txi/backend.py
@@ -74,7 +74,7 @@ def create_no_weights_model(self) -> None:
     def load_model_with_no_weights(self) -> None:
         original_volumes, self.config.volumes = (
             self.config.volumes,
-            {self.tmpdir.name: {"bind": "/data/hub/", "mode": "rw"}},
+            {self.tmpdir.name: {"bind": "/data", "mode": "rw"}},
         )
         self.load_model_from_pretrained()
         self.config.volumes = original_volumes