def METHOD_NAME(cls, inputs: Dict[str, Any]) -> Dict[str, Any]:
    with torch.no_grad():
        return cls.model.generate(inputs['inputs'],
                                  **inputs['forward_params'])