From acfc0606abeb1d44f86a07d898eef29e0d79b550 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 5 Feb 2025 05:24:36 -0800
Subject: [PATCH] Update rl.py

---
 unsloth/models/rl.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py
index cea08bbc3b..b041277e47 100644
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@@ -16,6 +16,7 @@ __all__ = [
     "PatchRL",
 ]
 
+import torch
 
 def PatchRL(FastLanguageModel):