File tree Expand file tree Collapse file tree 5 files changed +5
-5
lines changed Expand file tree Collapse file tree 5 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -262,7 +262,7 @@ def forward(
262262 key ,
263263 value ,
264264 attention_mask = attention_similarity ,
265- dropout = 0.0 if not self . training else self . dropout_p ,
265+ dropout = 0.0 ,
266266 scaling = self .scaling ,
267267 is_causal = self .is_causal ,
268268 ** kwargs ,
Original file line number Diff line number Diff line change @@ -893,7 +893,7 @@ def forward(
893893 key ,
894894 value ,
895895 attention_mask = attention_similarity ,
896- dropout = 0.0 if not self . training else self . dropout_p ,
896+ dropout = 0.0 ,
897897 scaling = self .scaling ,
898898 is_causal = self .is_causal ,
899899 ** kwargs ,
Original file line number Diff line number Diff line change @@ -940,7 +940,7 @@ def forward(
940940 key ,
941941 value ,
942942 attention_mask = attention_similarity ,
943- dropout = 0.0 if not self . training else self . dropout_p ,
943+ dropout = 0.0 ,
944944 scaling = self .scaling ,
945945 is_causal = self .is_causal ,
946946 ** kwargs ,
Original file line number Diff line number Diff line change @@ -477,7 +477,7 @@ def forward(
477477 key ,
478478 value ,
479479 attention_mask = attention_similarity ,
480- dropout = 0.0 if not self . training else self . dropout_p ,
480+ dropout = 0.0 ,
481481 scaling = self .scaling ,
482482 is_causal = self .is_causal ,
483483 ** kwargs ,
Original file line number Diff line number Diff line change @@ -674,7 +674,7 @@ def forward(
674674 key ,
675675 value ,
676676 attention_mask = attention_similarity ,
677- dropout = 0.0 if not self . training else self . dropout_p ,
677+ dropout = 0.0 ,
678678 scaling = self .scaling ,
679679 is_causal = self .is_causal ,
680680 ** kwargs ,
You can’t perform that action at this time.
0 commit comments