| from safetensors import safe_open | |
| import re | |
| path = "iska.safetensors" | |
| with safe_open(path, framework="pt") as f: | |
| meta = f.metadata() | |
| rank = meta.get("ss_network_dim") or meta.get("lora_rank") or meta.get("r") | |
| if rank: | |
| print("rank:", rank) | |
| else: | |
| # Look for a typical LoRA up/down weight | |
| up = next((k for k in f.keys() if re.search(r"lora(_\w+)?\.up\.weight$", k)), None) | |
| down = next((k for k in f.keys() if re.search(r"lora(_\w+)?\.down\.weight$", k)), None) | |
| if up: | |
| s = f.get_tensor(up).shape | |
| r = s[1] if len(s) in (2,4) else min(s) | |
| elif down: | |
| s = f.get_tensor(down).shape | |
| r = s[0] if len(s) in (2,4) else min(s) | |
| else: | |
| # Fallback: smallest dimension of any lora weight tensor | |
| k = next(k for k in f.keys() if "lora" in k and k.endswith(".weight")) | |
| s = f.get_tensor(k).shape | |
| r = min(s) | |
| print("rank:", r) |