Merge pull request axolotl-ai-cloud#272 from OpenAccess-AI-Collective…

…/model-revision support for loading a model by git revision
mkeoliya · Jul 14, 2023 · dd73b6a · dd73b6a
2 parents e428d5f + b62e8ec
commit dd73b6a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py
@@ -154,6 +154,8 @@ def load_model(
         )
 
     model_kwargs = {}
+    if cfg.model_revision:
+        model_kwargs["revision"] = cfg.model_revision
     if cfg.adapter == "qlora" and cfg.load_in_4bit:
         model_kwargs["quantization_config"] = BitsAndBytesConfig(
             load_in_4bit=True,