Ver Fonte

Create download_models.py

Xiaomeng Zhao há 1 ano atrás
pai
commit
7ce807f43f
1 ficheiros alterados com 4 adições e 0 exclusões
  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")