Stuck to finetuned Gemma 4 E4B for ComfYUI ( prompt enhancer )

[girzai]

Hi there, I finetuned Gemma 4 E4B with unsloth studio ( So I basically at the end of the training get the full model with all the files in the screenshot , I then convert the whole folder into a fp16 .safetensors compatible with ComfyUI ( script from Kijai https://huggingface.co/Comfy-Org/gemma-4/discussions/1 ) but when I use the model with any nodes it's crash my instance of ComfyUI . Is they're steps that I missed ? Thanks!