Upload README.md with huggingface_hub
Browse files
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(
|