transformers
bfaf9216 - Fix tests: 'Cohere2MoeModel' object has no attribute 'hf_device_map' (#46337)

Commit
45 days ago
Fix tests: 'Cohere2MoeModel' object has no attribute 'hf_device_map' (#46337) * cohere2_moe: skip invalid test cases Signed-off-by: kaixuanliu <kaixuan.liu@intel.com> * adjust the test case instead of just skip them Signed-off-by: kaixuanliu <kaixuan.liu@intel.com> * update `_TEXT_MODEL_TESTER_DEFAULTS` Signed-off-by: kaixuanliu <kaixuan.liu@intel.com> * Revert "update `_TEXT_MODEL_TESTER_DEFAULTS`" This reverts commit 538de3e5213052f9634f2aa85ec12d1143834eac. --------- Signed-off-by: kaixuanliu <kaixuan.liu@intel.com>
Author
Parents
Loading