dropout argument type error
#12
by
liangjs
- opened
The dropout_rate
argument passed to _flash_attention_forward
should be a float instead of a Dropout object.
https://huggingface.co/stabilityai/stable-code-3b/blob/0c6e452d83d5db540396651287bfbed4b966af7c/modeling_stablelm.py#L552-L561
Please check this: https://github.com/huggingface/transformers/pull/29236