cosimoiaia
commited on
Commit
·
5b298e9
1
Parent(s):
3d1e6ef
Update README.md
Browse files
README.md
CHANGED
@@ -50,8 +50,12 @@ from transformers import LLaMATokenizer, LLaMAForCausalLM, GenerationConfig
|
|
50 |
tokenizer = LLaMATokenizer.from_pretrained("cosimoiaia/Loquace-12B")
|
51 |
model = LLaMAForCausalLM.from_pretrained(
|
52 |
"cosimoiaia/Loquace-12B",
|
53 |
-
|
54 |
device_map="auto",
|
|
|
|
|
|
|
|
|
55 |
)
|
56 |
```
|
57 |
|
|
|
50 |
tokenizer = LLaMATokenizer.from_pretrained("cosimoiaia/Loquace-12B")
|
51 |
model = LLaMAForCausalLM.from_pretrained(
|
52 |
"cosimoiaia/Loquace-12B",
|
53 |
+
load_in_4bit=True,
|
54 |
device_map="auto",
|
55 |
+
quantization_config=BitsAndBytesConfig(
|
56 |
+
load_in_4bit=True,
|
57 |
+
llm_int8_has_fp16_weight=False
|
58 |
+
)
|
59 |
)
|
60 |
```
|
61 |
|