#DEBUG
import os
from safetensors.torch import load_file

lora_path = "lora_trained_model.safetensors"
if os.path.exists(lora_path):
    try:
        lora_data = load_file(lora_path)
        print("✅ LoRA file loaded successfully.")
        print("🔹 LoRA Keys:", list(lora_data.keys()))  # Check if keys match expected LoRA format
    except Exception as e:
        print(f"❌ Error reading LoRA file: {e}")