From 2518ea030ceb554766dc040a189229196127e2df Mon Sep 17 00:00:00 2001 From: shumingma Date: Thu, 8 Dec 2022 04:20:27 -0800 Subject: [PATCH] Fix example fsdp --- examples/fairseq/models/machine_translation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/fairseq/models/machine_translation.py b/examples/fairseq/models/machine_translation.py index 1e7761a..7c94ce8 100644 --- a/examples/fairseq/models/machine_translation.py +++ b/examples/fairseq/models/machine_translation.py @@ -11,7 +11,7 @@ from typing import Dict, List, Optional, Tuple import torch from fairseq import distributed_utils, utils -from fairseq.distributed import utils as fsdp_wrap +from fairseq.distributed import fsdp_wrap from fairseq.models import ( FairseqEncoder, FairseqEncoderDecoderModel,