import torch
import os


lora_path = "lora_trained_model.pt"

if os.path.exists(lora_path):
    try:
        state_dict = torch.load(lora_path, map_location="cpu")
        print("✅ LoRA checkpoint loaded successfully!")
        print("Keys in the checkpoint:", state_dict.keys())  # Print available keys
    except Exception as e:
        print(f"❌ Error loading LoRA checkpoint: {e}")
else:
    print("❌ LoRA file not found.")