Update ChatFalcon.py

ChatFalcon.py

@@ -33,10 +33,10 @@ from transformers.modeling_outputs import (
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import add_code_sample_docstrings, add_start_docstrings, add_start_docstrings_to_model_forward, logging
 from configuration_falcon import FalconConfig
-
+from transformers.models.auto import modeling_auto
 
 logger = logging.get_logger(__name__)
-
+modeling_auto.MODEL_FOR_CAUSAL_LM_MAPPING_NAMES['falcon'] = 'FalconForCausalLM'
 FALCON_PRETRAINED_MODEL_ARCHIVE_LIST = [
     "tiiuae/falcon-40b",
     "tiiuae/falcon-40b-instruct",