File tree Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -21,7 +21,7 @@ sphinxcontrib-serializinghtml
2121sphinx-autodoc-typehints == 1.11.1
2222pandas
2323einops
24- transformers >= 4.36.0 , < 4.41.0 ; python_version < = '3.10'
24+ transformers >= 4.53.0
2525mlflow >= 2.12.2
2626clearml >= 1.10.0rc0
2727tensorboardX
Original file line number Diff line number Diff 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." )
Original file line number Diff line number Diff line change @@ -35,7 +35,7 @@ tifffile; platform_system == "Linux" or platform_system == "Darwin"
3535pandas
3636requests
3737einops
38- transformers >= 4.36.0 , < 4.41.0 ; python_version < = '3.10'
38+ transformers >= 4.53.0
3939mlflow >= 2.12.2
4040clearml >= 1.10.0rc0
4141matplotlib >= 3.6.3
You can’t perform that action at this time.
0 commit comments