File size: 306 Bytes
6489817
 
 
 
 
1
2
3
4
5
6
#!/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": "./"})