Skip to content

Commit d90592a

Browse files
committed
added missing fields
1 parent f58b668 commit d90592a

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

sdks/python/apache_beam/examples/inference/pytorch_sentiment.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,16 @@ def _ensure_transformers_config_compat(config: DistilBertConfig) -> DistilBertCo
102102
config.torchscript = False
103103
if not hasattr(config, 'return_dict'):
104104
config.return_dict = True
105+
if not hasattr(config, 'output_attentions'):
106+
config.output_attentions = False
107+
if not hasattr(config, 'output_hidden_states'):
108+
config.output_hidden_states = False
109+
if not hasattr(config, 'use_cache'):
110+
config.use_cache = False
111+
if not hasattr(config, 'is_decoder'):
112+
config.is_decoder = False
113+
if not hasattr(config, 'add_cross_attention'):
114+
config.add_cross_attention = False
105115
return config
106116

107117

0 commit comments

Comments
 (0)