Kowsher commited on
Commit
1db61ef
·
1 Parent(s): 4fed768

Update configuration_falcon.py

Browse files
Files changed (1) hide show
  1. configuration_falcon.py +3 -1
configuration_falcon.py CHANGED
@@ -15,7 +15,7 @@
15
  """ Falcon configuration"""
16
  from transformers.configuration_utils import PretrainedConfig
17
  from transformers.utils import logging
18
-
19
 
20
  logger = logging.get_logger(__name__)
21
 
@@ -25,6 +25,8 @@ FALCON_PRETRAINED_CONFIG_ARCHIVE_MAP = {
25
  }
26
 
27
 
 
 
28
  class FalconConfig(PretrainedConfig):
29
  r"""
30
  This is the configuration class to store the configuration of a [`FalconModel`]. It is used to instantiate a Falcon
 
15
  """ Falcon configuration"""
16
  from transformers.configuration_utils import PretrainedConfig
17
  from transformers.utils import logging
18
+ from transformers.models.auto import modeling_auto
19
 
20
  logger = logging.get_logger(__name__)
21
 
 
25
  }
26
 
27
 
28
+ modeling_auto.MODEL_FOR_CAUSAL_LM_MAPPING_NAMES['falcon'] = 'FalconForCausalLM'
29
+
30
  class FalconConfig(PretrainedConfig):
31
  r"""
32
  This is the configuration class to store the configuration of a [`FalconModel`]. It is used to instantiate a Falcon