Fix user setting

openvinotoolkit · Aug 30, 2024 · 7fc1f59 · 7fc1f59
1 parent 0b9ed4a
commit 7fc1f59
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/otx/__init__.py b/src/otx/__init__.py
@@ -10,8 +10,12 @@
 
 from otx.core.types import *  # noqa: F403
 
-# Pretrained weight is saved into ~/.cache/torch/hub/checkpoints
-os.environ["XDG_CACHE_HOME"] = str(Path.home() / ".cache" / "torch" / "hub" / "checkpoints")
+# Set the value of XDG_CACHE_HOME to set the cache folder that stores the pretrained weights for timm and huggingface.
+# Default, Pretrained weight is saved into ~/.cache/torch/hub/checkpoints
+os.environ["XDG_CACHE_HOME"] = os.getenv(
+    "XDG_CACHE_HOME",
+    str(Path.home() / ".cache" / "torch" / "hub" / "checkpoints"),
+)
 
 OTX_LOGO: str = """