* add test for jamba with new model * ruff fix --------- Co-authored-by: Yehoshua Cohen <yehoshuaco@ai21.com>