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(
|