fix: compare inputs_embeds generate against input_ids generate, not u…#1787

akoumpa wants to merge 2 commits intomainNVIDIA-NeMo/Automodel:mainfrom

akoumparouli/fix_mamba_cached_vs_uncached_testNVIDIA-NeMo/Automodel:akoumparouli/fix_mamba_cached_vs_uncached_testCopy head branch name to clipboard