(config)
| 453 | |
| 454 | |
| 455 | def get_hf_model_path(config): |
| 456 | if config.model.path.startswith("hdfs:"): |
| 457 | from verl.utils.fs import copy_to_local |
| 458 | |
| 459 | local_model_path = copy_to_local(src=config.model.path, use_shm=config.model.get("use_shm", False)) |
| 460 | else: |
| 461 | local_model_path = config.model.path |
| 462 | return local_model_path |
| 463 | |
| 464 | |
| 465 | def load_megatron_model_weights(config, model_config, parallel_model, params_dtype, is_value_model=False): |
no test coverage detected