From 4765f2f621acfdcafa595409f8687d8fe4646ba0 Mon Sep 17 00:00:00 2001 From: David Corvoysier Date: Fri, 26 Jan 2024 14:58:45 +0100 Subject: [PATCH] Update optimum/neuron/utils/hub_neuronx_cache.py Co-authored-by: Michael Benayoun --- optimum/neuron/utils/hub_neuronx_cache.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/neuron/utils/hub_neuronx_cache.py b/optimum/neuron/utils/hub_neuronx_cache.py index b12dd9451..0eb484aa8 100644 --- a/optimum/neuron/utils/hub_neuronx_cache.py +++ b/optimum/neuron/utils/hub_neuronx_cache.py @@ -216,7 +216,7 @@ def hub_neuronx_cache(entry: Optional[CacheEntry] = None): """A context manager to activate the Hugging Face Hub proxy compiler cache. Args: - entry (`CacheEntry`): + entry (`Optional[CacheEntry]`, defaults to `None`): An optional dataclass containing metadata associated with the cache session. Will create a dedicated entry in the cache registry. """