chore: refactor truthy check and fix mypy (#780)

2023-10-24 12:28:40 +09:00
parent 6c81c61bc4
commit 11d1d607db
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/models.py
+++ b/src/axolotl/utils/models.py
@@ -31,7 +31,7 @@ LOG = logging.getLogger("axolotl")

 def load_model_config(cfg):
    model_config_name = cfg.base_model_config or cfg.base_model
-    trust_remote_code: bool = False or cfg.trust_remote_code
+    trust_remote_code = cfg.trust_remote_code is True
    return AutoConfig.from_pretrained(
        model_config_name, trust_remote_code=trust_remote_code
    )