Not working

#1
by bgeneto - opened

It doesn't work for me, someone else had success with it. I'm using unsloth gemma-4-E4B-it-UD-Q4_K_XL.gguf and gemma-4-E4B-it-assistant.Q8_0.gguf. llama cpp called this way:

CUDA_VISIBLE_DEVICES=1 ./build/bin/llama-server \
    --model ~/models/gemma-4-E4B-it-UD-Q4_K_XL.gguf \
    --temp 1.0 \
    --top-p 0.95 \
    --top-k 64 \
    --ctx-size 65536 \
    --flash-attn on \
    --mtp-head ~/models/gemma-4-E4B-it-assistant.Q8_0.gguf \
    --spec-type mtp \
    --draft-block-size 2 --draft-max 8 --draft-min 0 \
    -ngl 99 -ngld 99 \
    -ctk q4_0 -ctv q4_0 -ctkd q4_0 -ctvd q4_0 \
    --batch-size 1024 \
    --ubatch-size 1024 \
    --parallel 1 \
    --threads 8 \
    --threads-batch 8 \
    --cache-ram -1 \
    --no-mmap \
    --jinja \
    --host 0.0.0.0 \
    --port 8002

But no single request works, getting this error:

/app/ggml/src/ggml.c:3665: GGML_ASSERT(ggml_nelements(a) == ne0*ne1*ne2) failed

I had to export LD_LIBRARY_PATH=$(pwd) # run in your build/bin
And then it gets a little farther and the turbo3 quants start working.
Yet I find it is the --spect-type mtp that is still causing a crash.
And without it, there is no speedup over running just the model by itself...

The crash was caused by the RTX 3070's 860 flash attention heads
We're finding out more about this stuff every day.
Turboquant hasn't made flash attention templates for Ampere.
So it kills the server with an assertion.

It works by running the model with --flash-attn off
But then we're back at no speedup over running just the model by itself...
fattn.cu
146 case 512:
147 GGML_ASSERT(V->ne[0] == 512);
148 ggml_cuda_flash_attn_ext_mma_f16_switch_ncols2<512, 512>(ctx, dst);

This poem was brought to you by late night gdb session.

Sign up or log in to comment