gokaygokay
/

moondream-prompt

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

gokaygokay commited on Mar 26, 2024

Commit

1082928

·

1 Parent(s): c491b9d

repetition penalty

Files changed (1) hide show

moondream.py +3 -3

moondream.py CHANGED Viewed

@@ -64,7 +64,7 @@ class Moondream(PreTrainedModel):
         tokenizer,
         eos_text="<END>",
         max_new_tokens=128,
-        repetition_penalty=1.5,
         **kwargs,
     ):
         eos_tokens = tokenizer(eos_text, add_special_tokens=False)[0].ids
@@ -92,7 +92,7 @@ class Moondream(PreTrainedModel):
         question,
         tokenizer,
         chat_history="",
-        repetition_penalty=1.5,
         max_new_tokens=128,
         result_queue=None,
         **kwargs,
@@ -120,7 +120,7 @@ class Moondream(PreTrainedModel):
         images,
         prompts,
         tokenizer,
-        repetition_penalty=1.5,
         max_new_tokens=128,
         **kwargs,
     ):

         tokenizer,
         eos_text="<END>",
         max_new_tokens=128,
+        repetition_penalty=1.25,
         **kwargs,
     ):
         eos_tokens = tokenizer(eos_text, add_special_tokens=False)[0].ids
         question,
         tokenizer,
         chat_history="",
+        repetition_penalty=1.25,
         max_new_tokens=128,
         result_queue=None,
         **kwargs,
         images,
         prompts,
         tokenizer,
+        repetition_penalty=1.25,
         max_new_tokens=128,
         **kwargs,
     ):