Update rl.py

This commit is contained in:
Daniel Han 2025-02-14 04:31:27 -08:00
commit 4cce7ce242

View file

@ -21,6 +21,7 @@ from typing import Any, Callable, Dict, List, Literal, Optional, Tuple, Union
import inspect
import os
import re
import torch
from unsloth_zoo.compiler import create_new_function
from unsloth_zoo.logging_utils import PatchRLStatistics
from .rl_replacements import (