Marcus Cedric R. Idia commited on
Commit 路
01ffd01
1
Parent(s): ac6dbde
Update README.md
Browse files
README.md
CHANGED
|
@@ -26,9 +26,11 @@ This README provides instructions for running the Archimedes conversational AI a
|
|
| 26 |
Install requirements:
|
| 27 |
|
| 28 |
```
|
| 29 |
-
!pip install
|
| 30 |
-
!pip install
|
| 31 |
-
!pip install
|
|
|
|
|
|
|
| 32 |
```
|
| 33 |
|
| 34 |
## Usage
|
|
@@ -42,7 +44,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
|
|
| 42 |
login() # Need access to the gated model.
|
| 43 |
|
| 44 |
# Load LLAMA 2 model
|
| 45 |
-
model_name = "meta-llama/Llama-2-
|
| 46 |
|
| 47 |
# Quantization configuration
|
| 48 |
bnb_config = BitsAndBytesConfig(
|
|
|
|
| 26 |
Install requirements:
|
| 27 |
|
| 28 |
```
|
| 29 |
+
!pip install transformers
|
| 30 |
+
!pip install peft
|
| 31 |
+
!pip install torch
|
| 32 |
+
!pip install datasets
|
| 33 |
+
!pip install bitsandbytes
|
| 34 |
```
|
| 35 |
|
| 36 |
## Usage
|
|
|
|
| 44 |
login() # Need access to the gated model.
|
| 45 |
|
| 46 |
# Load LLAMA 2 model
|
| 47 |
+
model_name = "meta-llama/Llama-2-7b-chat-hf"
|
| 48 |
|
| 49 |
# Quantization configuration
|
| 50 |
bnb_config = BitsAndBytesConfig(
|