import torch

try:
    lora_checkpoint = torch.load("lora_trained_model.pt", map_location="cpu")
    print("Checkpoint loaded successfully!")
    print(type(lora_checkpoint))  # Print type to verify structure
except Exception as e:
    print(f"Error loading checkpoint: {e}")