Views
No views yet
1.
2├── model.py
3├── config.json
4├── vidhi_medassist.pt
5└── README.md1import torch
2from model import Vidhi, VidhiConfig
3
4device = "cuda"
5
6config = VidhiConfig()
7
8model = Vidhi(config)
9
10state_dict = torch.load(
11 "vidhi_medassist.pt",
12 map_location=device
13)
14
15model.load_state_dict(state_dict)
16
17model.to(device)
18model.eval()
19
20print("Vidhi loaded successfully")tiktoken)