Update _utils.py
This commit is contained in:
parent
4be7341e49
commit
67d40f3f6d
1 changed files with 3 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue