Skip to content

Commit 60155ff

Browse files
fix error with hidden state transform on rebase - implement abstract class method
Signed-off-by: Govind Ramnarayan <105831528+govind-ramnarayan@users.noreply.github.com>
1 parent d3bd853 commit 60155ff

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tensorrt_llm/_torch/auto_deploy/transform/library/hidden_states.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,11 @@ def get_global_buffer_initializers(cls, source_attn_node: Node) -> BufferInitial
258258
def get_constants(cls, source_attn_node: Node) -> List[Constant]:
259259
return []
260260

261+
@classmethod
262+
def get_standard_metadata_args(cls) -> List[str]:
263+
# unused, I think?
264+
return []
265+
261266

262267
@TransformRegistry.register("insert_cached_residual_add")
263268
class InsertCachedResidualAdd(InsertCachedAttention):

0 commit comments

Comments
 (0)