File size: 275 Bytes
70374ff
 
 
ada99f9
70374ff
 
 
 
1
2
3
4
5
6
7
8
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}")