acul3 commited on
Commit
e4ba5c0
1 Parent(s): 38e67b4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -3,11 +3,12 @@ license: bigscience-bloom-rail-1.0
3
  ---
4
 
5
  How to Use
6
-
 
7
  ```
8
  model_name_or_path = "bigscience/bloomz-7b1-mt"
9
 
10
- peft_model_id = f"acul3/bloomz-7b-instructions-lora"
11
 
12
  model = AutoModelForCausalLM.from_pretrained(model_name_or_path, device_map="auto")
13
 
 
3
  ---
4
 
5
  How to Use
6
+ from transformers import AutoModelForCausalLM, AutoTokenizer
7
+ from peft import PeftMode
8
  ```
9
  model_name_or_path = "bigscience/bloomz-7b1-mt"
10
 
11
+ peft_model_id = "acul3/bloomz-7b-instructions-lora"
12
 
13
  model = AutoModelForCausalLM.from_pretrained(model_name_or_path, device_map="auto")
14