Views
No views yet
1import torch 2ckpt = torch.load('dream_transformer.pt') 3model.load_state_dict(ckpt['model_state']) 4text = dream(model, temperature=1.3)