Переглянути джерело

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")