missing import

allenai · Apr 9, 2024 · 341cae2 · 341cae2
1 parent 1c8e127
commit 341cae2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hf_olmo/convert_olmo_to_hf.py b/hf_olmo/convert_olmo_to_hf.py
@@ -4,7 +4,7 @@
 import shutil
 
 import torch
-
+from omegaconf import OmegaConf as om
 from hf_olmo.configuration_olmo import OLMoConfig
 from hf_olmo.modeling_olmo import OLMoForCausalLM
 from hf_olmo.tokenization_olmo_fast import OLMoTokenizerFast