nytopop commited on
Commit
b987274
·
verified ·
1 Parent(s): a60493d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -7,6 +7,7 @@ base_model: huihui-ai/Llama-3.2-3B-Instruct-abliterated
7
  # creation
8
 
9
  ```python
 
10
  from transformers import AutoTokenizer, AutoModelForCausalLM
11
  from datasets import load_dataset
12
  from llmcompressor import oneshot
@@ -49,6 +50,7 @@ recipe = [
49
  model = AutoModelForCausalLM.from_pretrained(
50
  model_id,
51
  device_map="auto",
 
52
  )
53
 
54
  oneshot(
 
7
  # creation
8
 
9
  ```python
10
+
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
  from datasets import load_dataset
13
  from llmcompressor import oneshot
 
50
  model = AutoModelForCausalLM.from_pretrained(
51
  model_id,
52
  device_map="auto",
53
+ torch_dtype="bfloat16",
54
  )
55
 
56
  oneshot(