Update configuration_stacked.py
Browse files- configuration_stacked.py +7 -0
configuration_stacked.py
CHANGED
|
@@ -46,6 +46,13 @@ class ImpressoConfig(PretrainedConfig):
|
|
| 46 |
self.pretrained_config = pretrained_config
|
| 47 |
self.label_map = label_map
|
| 48 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
|
| 50 |
# Register the configuration with the transformers library
|
| 51 |
ImpressoConfig.register_for_auto_class()
|
|
|
|
| 46 |
self.pretrained_config = pretrained_config
|
| 47 |
self.label_map = label_map
|
| 48 |
|
| 49 |
+
@classmethod
|
| 50 |
+
def is_torch_support_available(cls):
|
| 51 |
+
"""
|
| 52 |
+
Indicate whether Torch support is available for this configuration.
|
| 53 |
+
Required for compatibility with certain parts of the Transformers library.
|
| 54 |
+
"""
|
| 55 |
+
return True
|
| 56 |
|
| 57 |
# Register the configuration with the transformers library
|
| 58 |
ImpressoConfig.register_for_auto_class()
|