* fix + test * remove transformer flax/torch warning * format
* tokenizer updates + tests * nit * add can_trim_prompt_cache * nits