From 0b994433b86efdf833dae073d8f1ce5537222bd0 Mon Sep 17 00:00:00 2001 From: Maanu Grover <109391026+maanug-nv@users.noreply.github.com> Date: Mon, 9 Dec 2024 13:38:20 -0800 Subject: [PATCH] Update Lightning import (#2549) * update import Signed-off-by: Maanu Grover * run formatting --------- Signed-off-by: Maanu Grover --- lm_eval/models/nemo_lm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lm_eval/models/nemo_lm.py b/lm_eval/models/nemo_lm.py index cf56019715..9b4c19e0ee 100644 --- a/lm_eval/models/nemo_lm.py +++ b/lm_eval/models/nemo_lm.py @@ -187,11 +187,11 @@ def __init__( **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: