emanuelaboros commited on
Commit
abef40f
1 Parent(s): 982a0e6

Upload model

Browse files
configuration_extended_multitask.py CHANGED
@@ -43,3 +43,7 @@ class ImpressoConfig(PretrainedConfig):
43
  self.use_cache = use_cache
44
  self.classifier_dropout = classifier_dropout
45
  self.pretrained_config = pretrained_config
 
 
 
 
 
43
  self.use_cache = use_cache
44
  self.classifier_dropout = classifier_dropout
45
  self.pretrained_config = pretrained_config
46
+
47
+
48
+ # Register the configuration with the transformers library
49
+ ImpressoConfig.register_for_auto_class()