Hugging Face's logo

Spaces:
big-vision
/
paligemma-hf
Running on Zero

App Files Community
2

Different results between Jax Space and the HF Transformers Space

#2
by Shalev - opened
Discussion
Shalev

From https://huggingface.co/spaces/big-vision/paligemma - the Jax model works well.
image.png

But the https://huggingface.co/spaces/big-vision/paligemma-hf space just selects the entire image (on the same input). I'm trying to reproduce the (better) Jax behavior on HF transformers, but I can't figure out what's being done differently on the Jax side. Any tips would be appreciated!

codelion

Seeing similar issues, is there a difference in the HF version?

merve

@Shalev @codelion I will debug and come back to you on this

D-Anel

Hi, how can we decode the segmentation tokens into binary mask for object segmentation?

codelion

@D-Anel you can check the code here - https://huggingface.co/spaces/big-vision/paligemma-hf/blob/main/app.py#L43

D-Anel

@codelion Thank you

D-Anel

@merve Did you find any solution on why the HF version does not perform? I am having the same issue as @Shalev but in segmentation. It would return a mas of zeros in HF version while works pretty well on jax.

Sign up or log in to comment