Update convert.py
Browse files- convert.py +1 -1
convert.py
CHANGED
|
@@ -10,7 +10,7 @@ logging.getLogger("torch.distributed.nn.jit.instantiator").setLevel(logging.WARN
|
|
| 10 |
from fairseq import checkpoint_utils
|
| 11 |
|
| 12 |
models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
|
| 13 |
-
["
|
| 14 |
)
|
| 15 |
model = models[0]
|
| 16 |
model.eval()
|
|
|
|
| 10 |
from fairseq import checkpoint_utils
|
| 11 |
|
| 12 |
models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
|
| 13 |
+
["checkpoint_best_legacy_500.pt"], suffix=""
|
| 14 |
)
|
| 15 |
model = models[0]
|
| 16 |
model.eval()
|