Unverified Commit 0b994433 authored by Maanu Grover's avatar Maanu Grover Committed by GitHub
Browse files

Update Lightning import (#2549)



* update import
Signed-off-by: default avatarMaanu Grover <maanug@nvidia.com>

* run formatting

---------
Signed-off-by: default avatarMaanu Grover <maanug@nvidia.com>
parent 2d11f2e5
......@@ -187,11 +187,11 @@ class NeMoLM(LM):
**kwargs,
):
try:
from lightning.pytorch.trainer.trainer import Trainer
from nemo.collections.nlp.modules.common.text_generation_utils import (
generate,
)
from nemo.collections.nlp.parts.nlp_overrides import NLPDDPStrategy
from pytorch_lightning.trainer.trainer import Trainer
self.generate = generate
except ModuleNotFoundError as exception:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment