ml-clara / openrlhf /models /__init__.py
dl3239491's picture
Upload folder using huggingface_hub
30c14cd verified
#
# For licensing see accompanying LICENSE file.
# Copyright (C) 2025 Apple Inc. All Rights Reserved.
#
from .actor import Actor
from .loss import (
DPOLoss,
GPTLMLoss,
KDLoss,
KTOLoss,
LogExpLoss,
PairWiseLoss,
PolicyLoss,
PRMLoss,
SFTLoss,
ValueLoss,
VanillaKTOLoss,
)
__all__ = [
"Actor",
"SFTLoss",
"DPOLoss",
"GPTLMLoss",
"KDLoss",
"KTOLoss",
"LogExpLoss",
"PairWiseLoss",
"PolicyLoss",
"PRMLoss",
"ValueLoss",
"VanillaKTOLoss",
]