vikhyatk
/

moondream2

Image-Text-to-Text

text-generation

Model card Files Files and versions

EthanReid commited on May 20

Commit

92d71f7

·

1 Parent(s): 782269c

build_text_model in init

Files changed (1) hide show

moondream.py +1 -1

moondream.py CHANGED Viewed

@@ -75,7 +75,7 @@ class MoondreamModel(nn.Module):
         self.vision = build_vision_model(config.vision, dtype)
-        self.text = None
         # Region Model
         self.region = nn.ModuleDict(

         self.vision = build_vision_model(config.vision, dtype)
+        self.text = build_text_model(config.text, torch.int8)
         # Region Model
         self.region = nn.ModuleDict(