Update ChatFalcon.py

ChatFalcon.py

@@ -22,7 +22,7 @@ import torch.utils.checkpoint
 from torch import nn
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, LayerNorm, MSELoss
 from torch.nn import functional as F
-
+from transformers import AutoModel,AutoModelForCausalLM, AutoConfig
 from transformers.modeling_outputs import (
     BaseModelOutputWithPastAndCrossAttentions,
     CausalLMOutputWithCrossAttentions,
@@ -1265,3 +1265,7 @@ class FalconForQuestionAnswering(FalconPreTrainedModel):
             hidden_states=outputs.hidden_states,
             attentions=outputs.attentions,
         )
+
+AutoConfig.register("falcon", FalconConfig)
+AutoModel.register(FalconConfig, FalconModel)
+AutoModelForCausalLM.register(FalconConfig, FalconForCausalLM)