fix: parse LORA keys dynamically instead of using fixed indices

Author: pl

transformers/llm/export/utils/lora.py

@@ -36,7 +36,8 @@ def load(self, path):
         with safe_open(path, framework="pt") as f:
             for k in f.keys():
                 names = k.split('.')
-                layer, key, name = names[4], names[6], names[7]
+                idx = names.index("layers")
+                layer, key, name = names[idx+1], names[idx+3], names[idx+4]
                 tag = layer + key
                 tensor = f.get_tensor(k).float()
                 self.lora_keys.add(key)