freeze (NVIDIA#7152)

Signed-off-by: arendu <adithya.r@gmail.com> Signed-off-by: dorotat <dorotat@nvidia.com>
dorotat-nv · Aug 24, 2023 · 90acab5 · 90acab5
1 parent 9541c22
commit 90acab5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nemo/collections/nlp/models/language_modeling/megatron_gpt_peft_models.py b/nemo/collections/nlp/models/language_modeling/megatron_gpt_peft_models.py
@@ -38,6 +38,7 @@ def __init__(self, cfg: DictConfig, trainer: Trainer):
         super().__init__(cfg, trainer)
         self.setup_complete = False
         self.base_keys = self.get_all_keys()
+        self.freeze()
         self.init_peft_modules()
         self.adapter_keys = self.get_all_keys() - self.base_keys