Add download support for modelscople.

keras_hub/src/utils/preset_utils.py

@@ -27,11 +27,12 @@
 KAGGLE_PREFIX = "kaggle://"
 GS_PREFIX = "gs://"
 HF_PREFIX = "hf://"
+MODELSCOPE_PREFIX = "modelscope://"
 
 KAGGLE_SCHEME = "kaggle"
 GS_SCHEME = "gs"
 HF_SCHEME = "hf"
-
+MODELSCOPE_SCHEME = "modelscope"
 ASSET_DIR = "assets"
 TOKENIZER_ASSET_DIR = f"{ASSET_DIR}/tokenizer"
 
@@ -165,6 +166,39 @@ def get_file(preset, path):
                 raise ValueError(message)
     elif scheme in tf_registered_schemes():
         return tf_copy_gfile_to_cache(preset, path)
+    elif scheme == MODELSCOPE_SCHEME:
+        try:
+            from modelscope.hub.snapshot_download import snapshot_download
+        except:
+            raise ImportError(
+                "`from_preset()` requires the `modelscope` package to "
+                "load from '{preset}'. "
+                "Please install with `pip install modelscope`."
+            )
+        modelscope_handle = preset.removeprefix(MODELSCOPE_SCHEME + "://")
+        try:
+            return_path =  snapshot_download(
+                modelscope_handle
+            )+'/'+path
+            if os.path.exists(return_path):
+                return return_path
+            raise FileNotFoundError(
+                    f"`{return_path}` doesn't exist in preset directory `{preset}`."
+                )
+        except HFValidationError as e:
+            raise ValueError(
+                "Unexpected ModelScope preset. Hugging Face model handles "
+                "should have the form 'modelscope://{org}/{model}'. For example, "
+                f"'modelscope://username/bert_base_en'. Received: preset={preset}."
+            ) from e
+        except EntryNotFoundError as e:
+            message = str(e)
+            if message.find("403 Client Error"):
+                raise FileNotFoundError(
+                    f"`{path}` doesn't exist in preset directory `{preset}`."
+                )
+            else:
+                raise ValueError(message)
     elif scheme == HF_SCHEME:
         if huggingface_hub is None:
             raise ImportError(