Views
No views yet
AutoModel.from_pretrained()
1from aes_model import AESModel 2import torch 3 4model = AESModel() 5model.load_state_dict(torch.load("pytorch_model.bin")) 6model.eval()