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()
|