Skip to content

Commit cc5de36

Browse files
authored
[Exaone4] Fixes the attn implementation! (#39906)
* fix * fix config
1 parent 00d4775 commit cc5de36

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/transformers/models/exaone4/configuration_exaone4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,7 @@ def __init__(
212212
for i in range(self.num_hidden_layers)
213213
]
214214
if "sliding_window" in self.layer_types:
215-
self._attn_implementation = "hybrid"
215+
self.cache_implementation = "hybrid"
216216
layer_type_validation(self.layer_types)
217217

218218
super().__init__(

src/transformers/models/exaone4/modular_exaone4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ def __init__(
246246
for i in range(self.num_hidden_layers)
247247
]
248248
if "sliding_window" in self.layer_types:
249-
self._attn_implementation = "hybrid"
249+
self.cache_implementation = "hybrid"
250250
layer_type_validation(self.layer_types)
251251

252252
super().__init__(

0 commit comments

Comments
 (0)