Kowsher
/

TokenTrails

Text Generation

text-generation-inference

Model card Files Files and versions

Kowsher commited on Jul 18, 2023

Commit

f212723

·

1 Parent(s): 3dc3074

Update ChatFalcon.py

Files changed (1) hide show

ChatFalcon.py +4 -0

ChatFalcon.py CHANGED Viewed

@@ -1271,3 +1271,7 @@ class FalconForQuestionAnswering(FalconPreTrainedModel):
 AutoConfig.register("falcon", FalconConfig)
 AutoModel.register(FalconConfig, FalconModel)
 AutoModelForCausalLM.register(FalconConfig, FalconForCausalLM)

 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")