Skip to content

Commit 8ab21be

Browse files
authored
enable static cache on vision encoder decoder (#39773)
Signed-off-by: jiqing-feng <[email protected]>
1 parent 67cfe11 commit 8ab21be

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@ def __init__(
113113

114114
self.encoder = encoder
115115
self.decoder = decoder
116+
self._can_compile_fullgraph = decoder._can_compile_fullgraph
116117

117118
if self.encoder.config.to_dict() != self.config.encoder.to_dict():
118119
logger.warning(

0 commit comments

Comments
 (0)