Hugging Face's logo

yifeihu
/
TFT-ID-1.0

Image-Text-to-Text
Transformers
Safetensors
florence2
vision
ocr
segmentation
custom_code
Model card Files
xet
 Community
2

Florence2LanguageForConditionalGeneration has no shift_tokens_right method defined in the class

#2
by sky-2002 - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+1
-1
sky-2002
•
edited Sep 8, 2024

Call prepare_decoder_input_ids_from_labels instead of shift_tokens_right as it is not defined in Florence2LanguageForConditionalGeneration
@yifeihu my code was breaking due to this, tried to fix, let me know if it works

 Florence2LanguageForConditionalGeneration has no shift_tokens_right method defined in the classb5b0b575
yifeihu
 Owner

@sky-2002 may i ask which transformers version are you using? I didn't encounter this issue when running the model. I'll look into it soon.
sky-2002

@yifeihu version is transformers==4.44.2
yifeihu
 Owner

@yifeihu version is transformers==4.44.2

@sky-2002 Hi I tried to replicate it but didn't encounter the error. I am also training a new version which I hope should be more stable.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment