Update ChatFalcon.py
Browse files- ChatFalcon.py +3 -0
ChatFalcon.py
CHANGED
|
@@ -1268,3 +1268,6 @@ class FalconForQuestionAnswering(FalconPreTrainedModel):
|
|
| 1268 |
)
|
| 1269 |
|
| 1270 |
|
|
|
|
|
|
|
|
|
|
|
|
| 1268 |
)
|
| 1269 |
|
| 1270 |
|
| 1271 |
+
AutoConfig.register("falcon", FalconConfig)
|
| 1272 |
+
AutoModel.register(FalconConfig, FalconModel)
|
| 1273 |
+
AutoModelForCausalLM.register(FalconConfig, FalconForCausalLM)
|