From 310f9ef6adcf1baf0eb0b369a04f1cc2431112cf Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Sun, 15 Dec 2024 17:41:04 +0100 Subject: [PATCH] test --- optimum_benchmark/backends/py_txi/backend.py | 2 +- optimum_benchmark/backends/py_txi/config.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/optimum_benchmark/backends/py_txi/backend.py b/optimum_benchmark/backends/py_txi/backend.py index 1b02277a..faf4f6bc 100644 --- a/optimum_benchmark/backends/py_txi/backend.py +++ b/optimum_benchmark/backends/py_txi/backend.py @@ -93,7 +93,7 @@ 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: - original_volumes, self.config.volumes = self.config.volumes, {self.tmpdir.name: {"bind": "/data", "mode": "rw"}} + original_volumes, self.config.volumes = self.config.volumes, {self.tmpdir.name: {"bind": "/data", "mode": "rwm"}} original_model, self.config.model = self.config.model, "/data/no_weights_model/" self.load_model_from_pretrained() self.config.model, self.config.volumes = original_model, original_volumes diff --git a/optimum_benchmark/backends/py_txi/config.py b/optimum_benchmark/backends/py_txi/config.py index 9cc23a64..20937af7 100644 --- a/optimum_benchmark/backends/py_txi/config.py +++ b/optimum_benchmark/backends/py_txi/config.py @@ -33,11 +33,11 @@ class PyTXIConfig(BackendConfig): metadata={"help": "Dictionary of ports to expose from the container."}, ) volumes: Dict[str, Any] = field( - default_factory=lambda: {HUGGINGFACE_HUB_CACHE: {"bind": "/data", "mode": "rw"}}, + default_factory=lambda: {HUGGINGFACE_HUB_CACHE: {"bind": "/data", "mode": "rwm"}}, metadata={"help": "Dictionary of volumes to mount inside the container."}, ) environment: List[str] = field( - default_factory=lambda: ["HUGGING_FACE_HUB_TOKEN"], + default_factory=lambda: ["HF_TOKEN"], metadata={"help": "List of environment variables to forward to the container from the host."}, )