Views
No views yet
1import torch
2from src.models.vae import create_vae # or appropriate model import
3
4# Load checkpoint
5checkpoint = torch.load("model.pt", map_location="cpu")
6
7# Create model and load weights
8model = create_model(...) # Use config from checkpoint
9model.load_state_dict(checkpoint["model_state_dict"])