Update configuration_falcon.py

configuration_falcon.py

@@ -15,7 +15,7 @@
 """ Falcon configuration"""
 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging
-
+from transformers.models.auto import modeling_auto
 
 logger = logging.get_logger(__name__)
 
@@ -25,6 +25,8 @@ FALCON_PRETRAINED_CONFIG_ARCHIVE_MAP = {
 }
 
 
+modeling_auto.MODEL_FOR_CAUSAL_LM_MAPPING_NAMES['falcon'] = 'FalconForCausalLM'
+
 class FalconConfig(PretrainedConfig):
     r"""
     This is the configuration class to store the configuration of a [`FalconModel`]. It is used to instantiate a Falcon