small fix in hf_config

mlfoundations · Apr 18, 2024 · 1004591 · 1004591
1 parent 07661b4
commit 1004591
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/.dockerignore b/.dockerignore
@@ -2,3 +2,18 @@ venv
 wandb
 logs
 checkpoints
+data
+results
+preproc_data
+logs
+wandb
+*.pt
+.pytest_cache
+.vscode
+.git
+tmp
+attention_logs
+not_val_shard
+attention_logs
+training/eval_data
+
diff --git a/open_lm/utils/transformers/hf_config.py b/open_lm/utils/transformers/hf_config.py
@@ -40,5 +40,5 @@ def __init__(
 
     def set_params(self, params: Params):
         self.tie_word_embeddings = params.weight_tying
-        for field in fields(params):
+        for field in fields(Params):
             setattr(self, field.name, getattr(params, field.name))