Skip to content

Commit 16a07af

Browse files
author
Sanggyu Lee
committed
Fix local-silent but CI-loud lint error
1 parent 70f1477 commit 16a07af

File tree

2 files changed

+4
-4
lines changed
  • test/modules/model/LlamaDecoderLayerWithKVCacheAndFusedAttention

2 files changed

+4
-4
lines changed

test/modules/model/LlamaDecoderLayerWithKVCacheAndFusedAttention/layer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,8 +126,8 @@ def forward_adapter(
126126
):
127127
# past_key_value is a dict with key_cache and value_cache.
128128
# It needs to be decomposed for tico and circle which does not know dict.
129-
key_cache = past_key_value.key_cache
130-
value_cache = past_key_value.value_cache
129+
key_cache = past_key_value.key_cache # type: ignore[union-attr]
130+
value_cache = past_key_value.value_cache # type: ignore[union-attr]
131131
return (
132132
torch.ops.circle.attention.llama(
133133
hidden_states,

test/modules/model/LlamaDecoderLayerWithKVCacheAndFusedAttention/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,8 +128,8 @@ def forward_adapter(
128128
):
129129
# past_key_value is a dict with key_cache and value_cache.
130130
# It needs to be decomposed for tico and circle which does not know dict.
131-
key_cache = past_key_value.key_cache
132-
value_cache = past_key_value.value_cache
131+
key_cache = past_key_value.key_cache # type: ignore[union-attr]
132+
value_cache = past_key_value.value_cache # type: ignore[union-attr]
133133
return (
134134
torch.ops.circle.attention.llama(
135135
hidden_states,

0 commit comments

Comments
 (0)