| { | |
| "load_config_kwargs": { | |
| "trust_remote_code": true | |
| }, | |
| "load_tokenizer_kwargs": { | |
| "pad_token": "<|endoftext|>", | |
| "trust_remote_code": true | |
| }, | |
| "finetune_args": {}, | |
| "mlflow_ft_conf": { | |
| "mlflow_hftransformers_misc_conf": { | |
| "config_hf_load_kwargs": { | |
| "trust_remote_code": true | |
| }, | |
| "tokenizer_hf_load_kwargs": { | |
| "return_token_type_ids": false | |
| }, | |
| "model_hf_load_kwargs": { | |
| "trust_remote_code": true, | |
| "ignore_mismatched_sizes": true | |
| }, | |
| "hf_predict_module": "phi_predict" | |
| }, | |
| "mlflow_save_model_kwargs": { | |
| "extra_pip_requirements": [ | |
| "einops" | |
| ] | |
| } | |
| } | |
| } |