danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add VLM model to download_models()

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-01-07 11:07:22 +02:00
parent 248eed8d7b
commit 2f413e937b
No known key found for this signature in database
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
haiku_rag_slim/haiku/rag/client.py
View file

@ -1672,6 +1672,9 @@ class HaikuRAG:
and self._config.reranking.model.provider == "ollama"
):
required_models.add(self._config.reranking.model.name)
pic_desc = self._config.processing.conversion_options.picture_description
if pic_desc.enabled and pic_desc.model.provider == "ollama":
required_models.add(pic_desc.model.name)
if not required_models:
return