Update ChatFalcon.py

ChatFalcon.py

@@ -23,6 +23,7 @@ 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.models.auto import modeling_auto
 from transformers.modeling_outputs import (
     BaseModelOutputWithPastAndCrossAttentions,
     CausalLMOutputWithCrossAttentions,
@@ -33,7 +34,7 @@ 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__)
 
@@ -1266,6 +1267,4 @@ class FalconForQuestionAnswering(FalconPreTrainedModel):
             attentions=outputs.attentions,
         )
 
-AutoConfig.register("falcon", FalconConfig)
-AutoModel.register(FalconConfig, FalconModel)
-AutoModelForCausalLM.register(FalconConfig, FalconForCausalLM)
+