Fix

huggingface · Feb 15, 2024 · 17c6f38 · 17c6f38
1 parent 3522de8
commit 17c6f38
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/optimum/commands/neuron/cache.py b/optimum/commands/neuron/cache.py
@@ -188,7 +188,7 @@ def parse_args(parser: "ArgumentParser"):
         parser.add_argument("--repo_id", type=str, default=None, help="The name of the repo to use as remote cache.")
 
     def _list_entries(self, mode: str):
-        entries = get_hub_cached_entries(mode, self.args.model_id, cache_repo_id=self.args.repo_id)
+        entries = get_hub_cached_entries(self.args.model_id, mode, cache_repo_id=self.args.repo_id)
         n_entries = len(entries)
         output = f"\n*** {n_entries} entrie(s) found in cache for {self.args.model_id} for {mode}.***\n\n"
         for entry in entries:

diff --git a/tests/cache/test_neuronx_cache.py b/tests/cache/test_neuronx_cache.py
@@ -123,7 +123,7 @@ def test_decoder_cache(cache_repos):
     synchronize_hub_cache(cache_repo_id=cache_repo_id)
     assert_local_and_hub_cache_sync(cache_path, cache_repo_id)
     # Verify we are able to fetch the cached entry for the model
-    model_entries = get_hub_cached_entries("inference", model_id, cache_repo_id=cache_repo_id)
+    model_entries = get_hub_cached_entries(model_id, "inference", cache_repo_id=cache_repo_id)
     assert len(model_entries) == 1
     assert model_entries[0] == model.config.neuron
     # Clear the local cache

diff --git a/text-generation-inference/server/text_generation_server/model.py b/text-generation-inference/server/text_generation_server/model.py
@@ -34,7 +34,7 @@ def get_export_kwargs_from_env():
 def is_cached(model_id, neuron_config):
     # Look for cached entries for the specified model
     in_cache = False
-    entries = get_hub_cached_entries("inference", model_id)
+    entries = get_hub_cached_entries(model_id, "inference")
     # Look for compatible entries
     for entry in entries:
         compatible = True