From f89ae2b2550f298ec8eedfe5483190d36b06ac6e Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 19 Feb 2025 23:43:52 -0800 Subject: [PATCH] vLLMSamplingParams --- unsloth/__init__.py | 1 + unsloth/models/rl.py | 1 + 2 files changed, 2 insertions(+) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index f0600f3328..ee3024bc99 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -210,6 +210,7 @@ except: pass from .models import * +from .rl import vLLMSamplingParams from .save import * from .chat_templates import * from .tokenizer_utils import * diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 572caf594c..0207f1c9bf 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -14,6 +14,7 @@ __all__ = [ "PatchFastRL", + "vLLMSamplingParams", ] import torch