update device-mode
Browse files- download_models_hf.py +1 -1
download_models_hf.py
CHANGED
|
@@ -60,7 +60,7 @@ if __name__ == '__main__':
|
|
| 60 |
json_mods = {
|
| 61 |
'models-dir': model_dir,
|
| 62 |
'layoutreader-model-dir': layoutreader_model_dir,
|
| 63 |
-
'device': 'cuda'
|
| 64 |
}
|
| 65 |
|
| 66 |
download_and_modify_json(json_url, config_file, json_mods)
|
|
|
|
| 60 |
json_mods = {
|
| 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)
|