class SkyworkRewardModel(BaseRewardModel):
None
)None
)auto
)flash_attention_2
)offload
)def __init__(
self,
model_type: Union[ModelType, str],
api_key: Optional[str] = None,
url: Optional[str] = None,
device_map: Optional[str] = 'auto',
attn_implementation: Optional[str] = 'flash_attention_2',
offload_folder: Optional[str] = 'offload'
):
def evaluate(self, messages: List[Dict[str, str]]):
def get_scores_types(self):