Adibvafa Fallahpour c269c5c74d Fix Mamba slow path bug with dtype mismatch. (#32691) ... * Fix Mamba slow path bug with dtype mismatch. * Update test_modeling_mamba.py * Improve style. * Fix issue with cache position of dtype mismatch test. * Change test for slow path. * Revert changes. * Switch to buggy code and add test to catch it. * Fix the dtype mismatch bug and add test code to verify it. * Fix minor bug with test. * Fix incorrect dtype of model output. * Fix incorrect dtype of cache. * Fix incorrect dtype of ssm cache. * Fix incorrect dtype of conv state. * Remove assertion for ssm state. * Add assertion for conv state dtype. * Fix all issues with dtype mismatch test.		2024-10-01 09:28:40 +02:00
..
__init__.py	[Add Mamba] Adds support for the Mamba models (#28094)	2024-03-05 20:01:06 +09:00
test_modeling_mamba.py	Fix Mamba slow path bug with dtype mismatch. (#32691)	2024-10-01 09:28:40 +02:00