Merge pull request #272 from OpenAccess-AI-Collective/model-revision

support for loading a model by git revision
axolotl-ai-cloud · Jul 14, 2023 · e1290d5 · e1290d5
2 parents fb48ad7 + 6130022
commit e1290d5
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,