Bug with some LLMs

[Mjall23]

Thanks to your optimizations, there's a significant increase in speed and performance !
But, I was also able to load the Gemma 2 2b parameters one. Now I can't load it, the app crashes if I try.
@Siddhesh2377