Views
No views yet
1from transformers import GPT2Model, GPT2Config
2
3# Load configuration
4config = GPT2Config.from_json_file('config.json')
5
6# Initialize model
7model = GPT2Model(config)
8model.load_state_dict(torch.load("model.pth"))
9
10# Example usage
11inputs = ...
12outputs = model(**inputs)