File size: 189 Bytes
275a1f6
 
 
 
 
 
 
1
2
3
4
5
6
7
8
import torch

# Load the Aiden_t5 model weights
aidenT5Model = torch.load('/path/to/model_weights/pytorch_model.bin')

# Initialize the Aiden_t5 model
aidenT5Model = Aiden_t5(aidenT5Model)