Skip to content

Commit bc37438

Browse files
authored
Signed-off-by: arendu <[email protected]>
1 parent 3a0f680 commit bc37438

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

nemo/collections/nlp/models/language_modeling/megatron_gpt_peft_models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ def __init__(self, cfg: DictConfig, trainer: Trainer):
3838
super().__init__(cfg, trainer)
3939
self.setup_complete = False
4040
self.base_keys = self.get_all_keys()
41+
self.freeze()
4142
self.init_peft_modules()
4243
self.adapter_keys = self.get_all_keys() - self.base_keys
4344

0 commit comments

Comments
 (0)