merge with main debug succeed

inst_follow/eval/bbh/run_eval.py

@@ -217,7 +217,7 @@ def main(args):
     load_from = args.load_from
 
     print("Loading model and tokenizer... from {}".format(load_from))
-
+    topic_router = None
     if load_from == "llama":
         ################# load model from llama  #############################
         model, tokenizer = load_from_llama()

mttl/config.py

@@ -8,7 +8,7 @@
 class Config:
     def __init__(self, filenames=None, kwargs=None, raise_error=True):
         # Stores personalization of the config file in a dict (json serializable)
-        self._updated_kwargs = {}
+        self._updated_kwargs = set()
         self.filenames = filenames
         self._set_defaults()
 
@@ -57,7 +57,9 @@ def update_kwargs(self, kwargs, eval=True, raise_error=True):
                 v = Template(v).substitute(os.environ)
 
             setattr(self, k, v)
-            self._updated_kwargs[k] = v
+            print(type(self._updated_kwargs))
+            # self._updated_kwargs[k] = v
+            self._updated_kwargs.add(k)
 
     def __getitem__(self, item):
         return getattr(self, item, None)
@@ -105,7 +107,7 @@ def parse(cls, extra_kwargs=None, raise_error=True):
 
         config = cls(args.config_files, args.kwargs, raise_error=raise_error)
 
-        print(config.to_json())
+        print("config", config.to_json())
         return config
 
     def _set_defaults(self):