Refix by casting to TransformerArgs

Author: Jack-Khuu

torchchat/cli/convert_hf_checkpoint.py

@@ -12,6 +12,8 @@
 
 import torch
 
+from torchchat.model import TransformerArgs
+
 # support running without installing as a package
 wd = Path(__file__).parent.parent
 sys.path.append(str(wd.resolve()))
@@ -32,8 +34,9 @@ def convert_hf_checkpoint(
     if model_name is None:
         model_name = model_dir.name
 
-    config = ModelArgs.from_name(model_name).transformer_args['text']
-    print(f"Model config {config}")
+    config_args = ModelArgs.from_name(model_name).transformer_args['text']
+    config = TransformerArgs.from_params(config_args)
+    print(f"Model config {config.__dict__}")
 
     # Load the json file containing weight mapping
     model_map_json = model_dir / "pytorch_model.bin.index.json"