Upload Moondream

configuration_moondream.py

@@ -1,8 +1,5 @@
 from transformers import PretrainedConfig
 
-from typing import Optional
-import math
-
 
 class PhiConfig(PretrainedConfig):
     model_type = "phi"

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:493ac8972766b8e4b9005bfab11454b93aab4987b44a01debebec3fa96773105
+oid sha256:15daa0c9a135d2084520e47d517ef689e09ee71c63d606bd8b8ff209bacc3e34
 size 3715037856

moondream.py

@@ -91,7 +91,7 @@ class Moondream(PreTrainedModel):
         result_queue=None,
         **kwargs,
     ):
-        prompt = f"<image>\n\n{chat_history}Question: {question}\n\nAnswer: "
+        prompt = f"<image>\n\n{chat_history}Question: {question}\n\nAnswer:"
         answer = self.generate(
             image_embeds,
             prompt,
@@ -120,7 +120,7 @@ class Moondream(PreTrainedModel):
         image_embeds = self.encode_image(images)
 
         templated_prompts = [
-            f"<image>\n\nQuestion: {prompt}\n\nAnswer: " for prompt in prompts
+            f"<image>\n\nQuestion: {prompt}\n\nAnswer:" for prompt in prompts
         ]
         prompt_embs = [
             self.input_embeds(prompt, image_embed.unsqueeze(0), tokenizer)[0]