Create modeling_stablelm_epoch.py
Browse files
modeling_stablelm_epoch.py
CHANGED
|
@@ -916,4 +916,4 @@ class StableLMEpochForCausalLM(StableLMEpochPreTrainedModel):
|
|
| 916 |
|
| 917 |
|
| 918 |
StableLMEpochConfig.register_for_auto_class()
|
| 919 |
-
StableLMEpochForCausalLM.register_for_auto_class("AutoModelForCausalLM")
|
|
|
|
| 916 |
|
| 917 |
|
| 918 |
StableLMEpochConfig.register_for_auto_class()
|
| 919 |
+
StableLMEpochForCausalLM.register_for_auto_class("AutoModelForCausalLM")
|