瀏覽代碼

Create download_models.py

Xiaomeng Zhao 1 年之前
父節點
當前提交
7ce807f43f
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      docs/download_models.py

+ 4 - 0
docs/download_models.py

@@ -0,0 +1,4 @@
+# use modelscope sdk download models
+from modelscope import snapshot_download
+model_dir = snapshot_download('wanderkid/PDF-Extract-Kit')
+print(f"model dir is: {model_dir}/models")