diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 4e03e0a168..6835f2e986 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -106,7 +106,7 @@ PRE_COMPILE_INFERENCE = [ "gpt_oss", ] -from transformers import GenerationConfig, CompileConfig, HybridCache, AutoConfig +from transformers import GenerationConfig, CompileConfig, AutoConfig try: from transformers import PreTrainedConfig @@ -117,8 +117,6 @@ except: HAS_TORCH_DTYPE = "torch_dtype" in PretrainedConfig.__doc__ -from transformers import GenerationConfig, CompileConfig, HybridCache - _compile_config = CompileConfig( fullgraph = False, dynamic = None,