Update download_models_hf.py
Browse files- download_models_hf.py +1 -0
download_models_hf.py
CHANGED
|
@@ -61,6 +61,7 @@ if __name__ == '__main__':
|
|
| 61 |
'models-dir': model_dir,
|
| 62 |
'layoutreader-model-dir': layoutreader_model_dir,
|
| 63 |
'device-mode': 'cuda'
|
|
|
|
| 64 |
}
|
| 65 |
|
| 66 |
download_and_modify_json(json_url, config_file, json_mods)
|
|
|
|
| 61 |
'models-dir': model_dir,
|
| 62 |
'layoutreader-model-dir': layoutreader_model_dir,
|
| 63 |
'device-mode': 'cuda'
|
| 64 |
+
# 'layout_model': 'doclayout_yolo'
|
| 65 |
}
|
| 66 |
|
| 67 |
download_and_modify_json(json_url, config_file, json_mods)
|