test

huggingface · Dec 16, 2024 · 307df81 · 307df81
1 parent e17e220
commit 307df81
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/optimum_benchmark/backends/py_txi/backend.py b/optimum_benchmark/backends/py_txi/backend.py
@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 from tempfile import TemporaryDirectory
 from typing import Any, Dict, List, Union
@@ -68,7 +69,7 @@ def create_no_weights_model(self) -> None:
     def load_model_with_no_weights(self) -> None:
         original_volumes, self.config.volumes = (
             self.config.volumes,
-            {Path(self.tmpdir.name) / "hub": {"bind": "/data", "mode": "rw"}},
+            {self.tmpdir.name: {"bind": "/data/hub/", "mode": "rw"}},
         )
         self.load_model_from_pretrained()
         self.config.volumes = original_volumes