Skip to content

Commit

Permalink
Update view model
Browse files Browse the repository at this point in the history
  • Loading branch information
moria97 committed Jun 4, 2024
1 parent f955786 commit 237901e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/pai_rag/app/web/view_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -313,11 +313,11 @@ def to_component_settings(self) -> Dict[str, Dict[str, Any]]:
"value": self.embed_model,
"visible": self.embed_source == "HuggingFace",
}
settings["embed_dim"] = (
{"value": EMBEDDING_DIM_DICT.get(self.embed_model, DEFAULT_EMBED_SIZE)}
settings["embed_dim"] = {
"value": EMBEDDING_DIM_DICT.get(self.embed_model, DEFAULT_EMBED_SIZE)
if self.embed_source == "HuggingFace"
else DEFAULT_EMBED_SIZE
)
}

settings["llm"] = {"value": self.llm}
settings["llm_eas_url"] = {"value": self.llm_eas_url}
Expand Down

0 comments on commit 237901e

Please sign in to comment.