Views
No views yet
1import torch
2from model.lila_e8 import LilaE8Config, LilaE8
3
4# Load checkpoint
5checkpoint = torch.load('checkpoint_step_9000.pt', map_location='cpu')
6config = checkpoint['config']
7model = LilaE8(config)
8model.load_state_dict(checkpoint['model_state_dict'])
9
10# Generate
11# (see inference examples in repo)