diff --git a/unsloth/models/__init__.py b/unsloth/models/__init__.py index 279080173f..b15e04ab74 100644 --- a/unsloth/models/__init__.py +++ b/unsloth/models/__init__.py @@ -20,4 +20,4 @@ from .mistral import FastMistralModel from .qwen2 import FastQwen2Model from .dpo import PatchDPOTrainer, PatchKTOTrainer from ._utils import is_bfloat16_supported -from .rl import PatchRL, PatchRLStatistics +from .rl import PatchFastRL