Add local base model loading (#315)

ludwig-ai · Oct 12, 2023 · 4826d48 · 4826d48
1 parent cbe92a3
commit 4826d48
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/docs/configuration/large_language_model.md b/docs/configuration/large_language_model.md
@@ -62,6 +62,13 @@ base_model: meta-llama/Llama-2-7b-hf
     these models are currently unsupported. If you have interest in using one of these models,
     please file a GitHub issue with your use case.
 
+You can also pass in a path to a locally saved Hugging Face model instead of loading from Hugging Face directly.
+
+Example:
+```yaml
+base_model: path/to/local/model/weights
+```
+
 ## HuggingFace Access Token
 
 Some base models like Llama-2 require authorization from HuggingFace to download,