From c1d028ced86aa4f7ad2ff5e89c5cac51209933a3 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 14 Feb 2025 04:32:24 -0800 Subject: [PATCH] Update rl_replacements.py --- unsloth/models/rl_replacements.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py index 39db053550..ed802e4877 100644 --- a/unsloth/models/rl_replacements.py +++ b/unsloth/models/rl_replacements.py @@ -19,6 +19,7 @@ __all__ = [ ] import re +import torch import inspect from collections import defaultdict RL_EXTRA_ARGS = defaultdict(list)