From 24d9963c8309c92560978ce639b5cdf89a2807b2 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 17 Feb 2025 20:08:15 -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 1b2f348541..f36598b0ac 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -94,6 +94,7 @@ from typing import * from dataclasses import dataclass, field from packaging.version import Version import torch +import numpy as np from contextlib import nullcontext from torch.nn import functional as F torch_compile_options = {{