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}") |