Initial commit

utils/lora_metadata.py

@@ -0,0 +1,16 @@
+from safetensors import safe_open
+from typing import Dict
+from .model_utils import determine_base_model
+
+async def extract_lora_metadata(file_path: str) -> Dict:
+    """Extract essential metadata from safetensors file"""
+    try:
+        with safe_open(file_path, framework="pt", device="cpu") as f:
+            metadata = f.metadata()
+            if metadata:
+                # Only extract base_model from ss_base_model_version
+                base_model = determine_base_model(metadata.get("ss_base_model_version"))
+                return {"base_model": base_model}
+    except Exception as e:
+        print(f"Error reading metadata from {file_path}: {str(e)}")
+    return {"base_model": "Unknown"}