From b891367829016adf7aacc063f09ec035de367afd Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 5 Feb 2025 15:11:40 -0800 Subject: [PATCH] Update __init__.py --- unsloth/models/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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