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
21
21
sphinx-autodoc-typehints == 1.11.1
22
22
pandas
23
23
einops
24
- transformers >= 4.36.0 , < 4.41.0 ; python_version < = '3.10'
24
+ transformers >= 4.53.0
25
25
mlflow >= 2.12.2
26
26
clearml >= 1.10.0rc0
27
27
tensorboardX
Original file line number Diff line number Diff line change @@ -333,6 +333,7 @@ def __init__(
333
333
"chunk_size_feed_forward" : chunk_size_feed_forward ,
334
334
"is_decoder" : is_decoder ,
335
335
"add_cross_attention" : add_cross_attention ,
336
+ "_attn_implementation" : "eager" ,
336
337
}
337
338
if not (0 <= drop_out <= 1 ):
338
339
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"
35
35
pandas
36
36
requests
37
37
einops
38
- transformers >= 4.36.0 , < 4.41.0 ; python_version < = '3.10'
38
+ transformers >= 4.53.0
39
39
mlflow >= 2.12.2
40
40
clearml >= 1.10.0rc0
41
41
matplotlib >= 3.6.3
You can’t perform that action at this time.
0 commit comments