import torch

# Specify the correct path to your LoRA checkpoint file
lora_path = "./lora_trained_model.pt"  # Replace with the correct file name

# Define the device (use CPU in this case)
device = torch.device("cpu")

# Load the checkpoint
checkpoint = torch.load(lora_path, map_location=device)

# Print the keys in the checkpoint to check the structure
print("Checkpoint keys:", checkpoint.keys())