From 46cc4aa9d7ff5e0bfaf8b2484c84c5b9b1aab8fb Mon Sep 17 00:00:00 2001 From: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com> Date: Fri, 11 Oct 2024 14:33:52 +0200 Subject: [PATCH] Update optimum/exporters/onnx/model_configs.py --- optimum/exporters/onnx/model_configs.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index d31fef36cf5..6e948e1ad1e 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -265,10 +265,6 @@ class DecisionTransformerOnnxConfig(GPT2OnnxConfig): @property def inputs(self) -> Dict[str, Dict[int, str]]: - DEFAULT_DUMMY_SHAPES['actions'] = self._normalized_config.config.act_dim - DEFAULT_DUMMY_SHAPES['states'] = self._normalized_config.config.state_dim - DEFAULT_DUMMY_SHAPES['returns'] = 1 - DEFAULT_DUMMY_SHAPES['last_hidden_state'] = self._normalized_config.config.hidden_size return { 'states': {0: 'batch_size', 1: 'sequence_length', 2: 'states'},