Question Answering
PEFT
English
Marcus Cedric R. Idia commited on
Commit
01ffd01
1 Parent(s): ac6dbde

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -4
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 huggingface
30
- !pip install -q -U trl transformers accelerate git+https://github.com/huggingface/peft.git
31
- !pip install -q datasets bitsandbytes einops wandb
 
 
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-13b-chat-hf"
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(