#!/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": "./"}) | |