irotem98
/

edge_vlm

irotem98 commited on 6 days ago

Commit

8fe453b

•

1 Parent(s): 13a7b44

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -112,7 +112,7 @@ class MoondreamModel(nn.Module):
     @staticmethod
-    def generate_caption(model, image, tokenizer, max_length=128):
         model.eval()
         with torch.no_grad():
             image = image.unsqueeze(0).to(DEVICE)

     @staticmethod
+    def generate_caption(model, image, tokenizer, max_length=192):
         model.eval()
         with torch.no_grad():
             image = image.unsqueeze(0).to(DEVICE)