ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2025-06-30 23:13:33 -07:00
commit ce923e31dc
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/_utils.py
View file

@ -77,7 +77,7 @@ import contextlib
import re
import warnings, subprocess, re, inspect, psutil, os, math
from unsloth_zoo.utils import Version
from unsloth import DEVICE_TYPE
from unsloth_zoo import DEVICE_TYPE
from unsloth_zoo.tokenizer_utils import (
patch_tokenizer as _patch_tokenizer,
@ -553,8 +553,6 @@ UNSLOTH_COMPILE_IGNORE_ERRORS = os.environ.get("UNSLOTH_COMPILE_IGNORE_ERRORS",
import functools
from torch._inductor.runtime.hints import DeviceProperties
from unsloth import DEVICE_TYPE
@functools.lru_cache(None)
def is_big_gpu(index) -> bool: