come non detto
Browse files- __init__.py +1 -1
__init__.py
CHANGED
|
@@ -6,6 +6,6 @@ AutoConfig.register("stldec", STLConfig)
|
|
| 6 |
AutoModel.register(STLConfig, STLModel)
|
| 7 |
AutoModelForCausalLM.register(STLConfig, STLForCausalLM)
|
| 8 |
|
| 9 |
-
|
| 10 |
STLModel.register_for_auto_class("AutoModel")
|
| 11 |
STLForCausalLM.register_for_auto_class("AutoModelForCausalLM")
|
|
|
|
| 6 |
AutoModel.register(STLConfig, STLModel)
|
| 7 |
AutoModelForCausalLM.register(STLConfig, STLForCausalLM)
|
| 8 |
|
| 9 |
+
STLConfig.register_for_auto_class()
|
| 10 |
STLModel.register_for_auto_class("AutoModel")
|
| 11 |
STLForCausalLM.register_for_auto_class("AutoModelForCausalLM")
|