blackhole33 commited on
Commit
8d8f1a9
1 Parent(s): 1aa9f65

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -1
README.md CHANGED
@@ -22,7 +22,20 @@ It achieves the following results on the evaluation set:
22
 
23
  ## Model description
24
 
25
- More information needed
 
 
 
 
 
 
 
 
 
 
 
 
 
26
 
27
  ## Intended uses & limitations
28
 
 
22
 
23
  ## Model description
24
 
25
+ ```
26
+
27
+ MODEL_NAME = "/content/blackhole33/llama-5000-sample-peft"
28
+ quantization_config = BitsAndBytesConfig(
29
+ load_in_4bit=True, bnb_4bit_quant_type="nf4", bnb_4bit_compute_dtype=torch.bfloat16
30
+ )
31
+
32
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_fast=True)
33
+
34
+ model = AutoModelForCausalLM.from_pretrained(
35
+ MODEL_NAME, quantization_config=quantization_config, device_map="auto"
36
+ )
37
+
38
+ ```
39
 
40
  ## Intended uses & limitations
41