Marcus Cedric R. Idia
commited on
Commit
•
74f6a3a
1
Parent(s):
75616c9
Update README.md
Browse files
README.md
CHANGED
|
@@ -44,7 +44,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
|
|
| 44 |
login() # Need access to the gated model.
|
| 45 |
|
| 46 |
# Load LLAMA 2 model
|
| 47 |
-
model_name = "meta-llama/Llama-2-
|
| 48 |
|
| 49 |
# Quantization configuration
|
| 50 |
bnb_config = BitsAndBytesConfig(
|
|
|
|
| 44 |
login() # Need access to the gated model.
|
| 45 |
|
| 46 |
# Load LLAMA 2 model
|
| 47 |
+
model_name = "meta-llama/Llama-2-13b-chat-hf"
|
| 48 |
|
| 49 |
# Quantization configuration
|
| 50 |
bnb_config = BitsAndBytesConfig(
|