fix

Author: aIbrahiim

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

@@ -96,22 +96,21 @@ def _ensure_transformers_config_compat(config: DistilBertConfig) -> DistilBertCo
   The benchmark can run with container images whose transformers version differs
   from the launcher environment. Some versions assume these attributes exist.
   """
+  # Use a default config instance as the source of canonical attributes for the
+  # transformers version available on the worker. This avoids chasing one
+  # missing field at a time (e.g. torchscript, output_attentions).
+  default_config = DistilBertConfig()
+  for key, value in default_config.to_dict().items():
+    if not hasattr(config, key):
+      setattr(config, key, value)
+
+  # Keep non-serialized fields explicitly for older/newer transformers mixes.
   if not hasattr(config, 'pruned_heads'):
     config.pruned_heads = {}
   if not hasattr(config, 'torchscript'):
     config.torchscript = False
   if not hasattr(config, 'return_dict'):
     config.return_dict = True
-  if not hasattr(config, 'output_attentions'):
-    config.output_attentions = False
-  if not hasattr(config, 'output_hidden_states'):
-    config.output_hidden_states = False
-  if not hasattr(config, 'use_cache'):
-    config.use_cache = False
-  if not hasattr(config, 'is_decoder'):
-    config.is_decoder = False
-  if not hasattr(config, 'add_cross_attention'):
-    config.add_cross_attention = False
   return config