Update moondream.py
Browse files- moondream.py +1 -1
moondream.py
CHANGED
@@ -84,7 +84,7 @@ class Moondream(PreTrainedModel):
|
|
84 |
output_ids = self.text_model.generate(
|
85 |
inputs_embeds=inputs_embeds, streamer=streamer, **generate_config
|
86 |
)
|
87 |
-
|
88 |
|
89 |
return tokenizer.batch_decode(output_ids, skip_special_tokens=True)
|
90 |
|
|
|
84 |
output_ids = self.text_model.generate(
|
85 |
inputs_embeds=inputs_embeds, streamer=streamer, **generate_config
|
86 |
)
|
87 |
+
|
88 |
|
89 |
return tokenizer.batch_decode(output_ids, skip_special_tokens=True)
|
90 |
|