saracandu commited on
Commit
bdedfcc
·
verified ·
1 Parent(s): 0b21222

come non detto

Browse files
Files changed (1) hide show
  1. __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
- AutoConfig.register_for_auto_class()
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")