use different model cache than default

nebula-contrib · Jan 20, 2024 · 14feb12 · 14feb12
1 parent 231e152
commit 14feb12
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/nebulagraph_lite/nebulagraph.py b/src/nebulagraph_lite/nebulagraph.py
@@ -140,11 +140,17 @@ def _try_download_modelscope(self):
         try:
             from modelscope.hub.file_download import model_file_download
 
+            # create cache folder
+            cache_path = f"{self.base_path}/cache"
+            os.makedirs(cache_path, exist_ok=True)
+            os.environ["MODELSCOPE_CACHE"] = cache_path
+
             # download nebulagraph_lite image tarball
             model_file = model_file_download(
                 model_id=MODELSCOPE_MODEL_ID,
                 file_path=MODELSCOPE_MODEL_FILE_PATH,
                 revision=MODELSCOPE_MODEL_VERSION,
+                target_path=self.base_path,
             )
             # download udocker tarball
             tarball_file = model_file_download(