File size: 306 Bytes
17530a7 |
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": "./"})
|