Fix conversion bug (#116)

* Fix

* Format

Author: yeandy

convert_checkpoints.py

@@ -179,7 +179,10 @@ def _merge_llama_weights(
         f"{len(tensors)} shards (shape = {tensors[0].shape}) for {key})"
     )
     state_dict_for_key = {}
-    for pattern, kind in llama_model.get_weight_sharding_type.items():
+    weight_sharding_type = (
+        llama_model.Transformer.get_weight_sharding_type().items()
+    )
+    for pattern, kind in weight_sharding_type:
       if not key.endswith(pattern):
         continue
       with torch.no_grad():