Fix

huggingface · Feb 15, 2024 · 3522de8 · 3522de8
1 parent 738ef38
commit 3522de8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/optimum/commands/neuron/cache.py b/optimum/commands/neuron/cache.py
@@ -21,8 +21,6 @@
     CACHE_REPO_NAME,
     HF_HOME_CACHE_REPO_FILE,
     create_custom_cache_repo,
-    list_in_registry,
-    load_custom_cache_repo_name_from_hf_home,
     set_custom_cache_repo_name_in_hf_home,
 )
 from ...neuron.utils.runner import ExampleRunner
@@ -226,7 +224,7 @@ class CustomCacheRepoCommand(BaseOptimumCLICommand):
         ),
         CommandInfo(
             name="synchronize",
-            help="Synchronize the neuronx compiler cache with a hub cache repo (inferentia only).",
+            help="Synchronize the neuronx compiler cache with a hub cache repo.",
             subcommand_class=SynchronizeRepoCommand,
         ),
         CommandInfo(