Upload Moondream

moondream.py

@@ -12,7 +12,7 @@ class Moondream(PreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
         self.vision_encoder = VisionEncoder()
-        self.text_model = TextModel(config, None)
+        self.text_model = TextModel(config)
 
     @property
     def device(self):

text_model.py

@@ -7,7 +7,7 @@ transformers.logging.set_verbosity_error()
 
 
 class TextModel(nn.Module):
-    def __init__(self, config, tokenizer) -> None:
+    def __init__(self, config) -> None:
         super().__init__()
 
         if type(config.phi_config) == dict:
@@ -17,4 +17,3 @@ class TextModel(nn.Module):
 
         self.model = PhiForCausalLM(phi_config)
         self.text_emb = self.model.get_input_embeddings()
-        self.tokenizer = tokenizer