| #!/usr/bin/env python3 | |
| from fairseq.checkpoint_utils import load_model_ensemble_and_task | |
| # model = load_model_ensemble_and_task(["./pytorch_model.pt"], arg_overrides={"config_yaml": "./config.yaml", "data": "./"}) | |
| model = load_model_ensemble_and_task(["./pytorch_model.pt"], arg_overrides={"data": "./"}) | |