1import torch
2from huggingface_hub import hf_hub_download
3
4ruta = hf_hub_download("DarksitoBest/minigpt-casual-es", "pretrain.pt")
5checkpoint = torch.load(ruta, map_location="cpu")
6
7model = MiniGPT(n_heads=8, n_layers=6)
8model.load_state_dict(checkpoint['model_state'])