map_location=device, weights_only=True, )model = cls(config) model.load_state_dict(state_dict) model.to(device) model.device = device model.eval()for param in model.parameters(): param.requires_grad = Falsereturn model 3 changes: 3 additions & 0 deletions 3 backend/shared_configs/configs...