Kowsher
/

TokenTrails

Text Generation

text-generation-inference

Model card Files Files and versions

Kowsher commited on Jul 18, 2023

Commit

231705d

·

1 Parent(s): 9fec138

Update ChatFalcon.py

Files changed (1) hide show

ChatFalcon.py +3 -2

ChatFalcon.py CHANGED Viewed

@@ -1267,11 +1267,12 @@ class FalconForQuestionAnswering(FalconPreTrainedModel):
             attentions=outputs.attentions,
         )
 AutoConfig.register("falcon", FalconConfig)
 AutoModel.register(FalconConfig, FalconModel)
 AutoModelForCausalLM.register(FalconConfig, FalconForCausalLM)
 FalconConfig.register_for_auto_class()
 FalconModel.register_for_auto_class("AutoModel")
-FalconForCausalLM.register_for_auto_class("AutoModelForCausalLM")

             attentions=outputs.attentions,
         )
+'''
 AutoConfig.register("falcon", FalconConfig)
 AutoModel.register(FalconConfig, FalconModel)
 AutoModelForCausalLM.register(FalconConfig, FalconForCausalLM)
 FalconConfig.register_for_auto_class()
 FalconModel.register_for_auto_class("AutoModel")
+FalconForCausalLM.register_for_auto_class("AutoModelForCausalLM")
+'''