Skip to content

Commit

Permalink
Config: Fetch the correct dict for draft_model and lora
Browse files Browse the repository at this point in the history
Fixed fetching from the merged config instead of the sub-config

Signed-off-by: kingbri <bdashore3@proton.me>
  • Loading branch information
kingbri1 committed Sep 11, 2024
1 parent aa832b8 commit 62beb2b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions common/tabby_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,8 @@ def load(self, arguments: Optional[dict] = None):
self.network = unwrap(merged_config.get("network"), {})
self.logging = unwrap(merged_config.get("logging"), {})
self.model = unwrap(merged_config.get("model"), {})
self.draft_model = unwrap(merged_config.get("draft"), {})
self.lora = unwrap(merged_config.get("draft"), {})
self.draft_model = unwrap(self.model.get("draft"), {})
self.lora = unwrap(self.model.get("lora"), {})
self.sampling = unwrap(merged_config.get("sampling"), {})
self.developer = unwrap(merged_config.get("developer"), {})
self.embeddings = unwrap(merged_config.get("embeddings"), {})
Expand Down

0 comments on commit 62beb2b

Please sign in to comment.