stepfun-ai
/

GOT-OCR2_0

Image-Text-to-Text

feature-extraction

vision-language

Model card Files Files and versions Community

ucaslcl commited on Sep 13, 2024

Commit

ffd1620

·

verified ·

1 Parent(s): d18cb8e

Update modeling_GOT.py

Files changed (1) hide show

modeling_GOT.py +1 -1

modeling_GOT.py CHANGED Viewed

@@ -570,7 +570,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
                 do_sample=False,
                 num_beams = 1,
                 no_repeat_ngram_size = 20,
-                streamer=streamer,
                 max_new_tokens=4096,
                 stopping_criteria=[stopping_criteria]
                 )

                 do_sample=False,
                 num_beams = 1,
                 no_repeat_ngram_size = 20,
+                # streamer=streamer,
                 max_new_tokens=4096,
                 stopping_criteria=[stopping_criteria]
                 )