Update README.md
Browse files
README.md
CHANGED
|
@@ -27,5 +27,5 @@ This llama model was trained 2x faster with [Unsloth](https://github.com/unsloth
|
|
| 27 |
|
| 28 |
|
| 29 |
# Load model directly
|
| 30 |
-
|
| 31 |
-
model = AutoModel.from_pretrained("AdityaKothari/WellnessAI-8B")
|
|
|
|
| 27 |
|
| 28 |
|
| 29 |
# Load model directly
|
| 30 |
+
from transformers import AutoModel
|
| 31 |
+
model = AutoModel.from_pretrained("AdityaKothari/WellnessAI-8B")
|