Skip to content

Commit b52edd3

Browse files
committed
Possible fix for transformers version
Signed-off-by: Eric Kerfoot <[email protected]>
1 parent 0701945 commit b52edd3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

monai/networks/nets/transchex.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -333,6 +333,7 @@ def __init__(
333333
"chunk_size_feed_forward": chunk_size_feed_forward,
334334
"is_decoder": is_decoder,
335335
"add_cross_attention": add_cross_attention,
336+
"_attn_implementation": "eager",
336337
}
337338
if not (0 <= drop_out <= 1):
338339
raise ValueError("dropout_rate should be between 0 and 1.")

0 commit comments

Comments
 (0)