blewis-hir commited on
Commit
277a89d
·
verified ·
1 Parent(s): 2b88640

Fix incorrect version check

Browse files
Files changed (1) hide show
  1. modeling_decilm.py +2 -2
modeling_decilm.py CHANGED
@@ -32,7 +32,7 @@ from transformers import GenerationConfig
32
  from transformers.generation.utils import GenerationMixin, GenerateOutput
33
 
34
 
35
- if Version(importlib.metadata.version("transformers")) >= Version("4.56.0.dev0")
36
  from transformers.generation.configuration_utils import NEED_SETUP_CACHE_CLASSES_MAPPING
37
  else:
38
  from transformers.generation.configuration_utils import ALL_STATIC_CACHE_IMPLEMENTATIONS
@@ -818,7 +818,7 @@ class DeciLMPreTrainedModel(PreTrainedModel):
818
  # DeciLM-specific code
819
  generation_config, model_kwargs = super()._prepare_generation_config(generation_config, *args, **kwargs)
820
  generation_config.cache_implementation = "variable"
821
- if transformers_version >= Version("4.56.0.dev0")
822
  NEED_SETUP_CACHE_CLASSES_MAPPING["variable"] = VariableCache
823
  else:
824
  ALL_STATIC_CACHE_IMPLEMENTATIONS["variable"] = VariableCache
 
32
  from transformers.generation.utils import GenerationMixin, GenerateOutput
33
 
34
 
35
+ if Version(importlib.metadata.version("transformers")) <= Version("4.56.0.dev0")
36
  from transformers.generation.configuration_utils import NEED_SETUP_CACHE_CLASSES_MAPPING
37
  else:
38
  from transformers.generation.configuration_utils import ALL_STATIC_CACHE_IMPLEMENTATIONS
 
818
  # DeciLM-specific code
819
  generation_config, model_kwargs = super()._prepare_generation_config(generation_config, *args, **kwargs)
820
  generation_config.cache_implementation = "variable"
821
+ if transformers_version <= Version("4.56.0.dev0")
822
  NEED_SETUP_CACHE_CLASSES_MAPPING["variable"] = VariableCache
823
  else:
824
  ALL_STATIC_CACHE_IMPLEMENTATIONS["variable"] = VariableCache