Update _utils.py

This commit is contained in:
Daniel Han 2024-11-20 19:40:08 -08:00
commit 67d40f3f6d

View file

@ -1111,7 +1111,9 @@ def unsloth_compile_transformers(
cudagraphs = False,
debug = False,
import_from_cache = False,
disable = False,
):
if disable: return
model_types = get_transformers_model_type(
model_name = model_name,
token = token,
@ -1140,6 +1142,7 @@ def unsloth_compile_transformers(
cudagraphs = cudagraphs,
debug = debug,
import_from_cache = import_from_cache,
disable = disable,
)
pass
return