emanuelaboros commited on
Commit
8560bf5
·
verified ·
1 Parent(s): 97a98d1

Update configuration_stacked.py

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