colin
/
Witllm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix light module config error.

This commit is contained in:
Colin 2025-08-13 22:08:21 +08:00
parent 59e079f5e7
commit 57a06fed27
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wit/model/light_module.py
View File

@ -173,7 +173,7 @@ class LightModule(pl.LightningModule):
use_tril_attention_mask = conf.use_tril_attention_mask use_tril_attention_mask = conf.use_tril_attention_mask
super().__init__() super().__init__()
self.automatic_optimization = False self.automatic_optimization = False
self.save_hyperparameters() self.save_hyperparameters(ignore="model")
if mconf == None: if mconf == None:
mconf = ModelConfig() mconf = ModelConfig()
if pretrained_model_dir != None: if pretrained_model_dir != None: