From 88126e33db9f9153d22ba5c56510c138258267cd Mon Sep 17 00:00:00 2001 From: Alexandros Koumparoulis Date: Tue, 13 Aug 2024 16:26:54 -0700 Subject: [PATCH] Remove unused class Signed-off-by: Alexandros Koumparoulis --- nemo/lightning/pytorch/plugins/precision.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/nemo/lightning/pytorch/plugins/precision.py b/nemo/lightning/pytorch/plugins/precision.py index 9bc5992c77b0..93b8516b7048 100644 --- a/nemo/lightning/pytorch/plugins/precision.py +++ b/nemo/lightning/pytorch/plugins/precision.py @@ -88,15 +88,6 @@ def make_default_dtype_config_from_precision(precision: str): ), f"Error: Expected precision to be FP32, BF16 or FP16, {precision}" return config - -class McoreConfigsFromPlugin: - def generate_ddp_config(self): - return {} - def generate_optim_config(self): - return {} - def generate_model_config(self): - return {} - class MegatronCustomPrecision(MixedPrecision, McoreConfigsFromPlugin): def __init__( self,