Commit
•
79ea4d2
1
Parent(s):
e842e68
Update README.md
Browse files
README.md
CHANGED
@@ -44,12 +44,15 @@ https://huggingface.co/cosimoiaia/Loquace-20B - Based on gpt-neox-20B
|
|
44 |
|
45 |
|
46 |
```python
|
47 |
-
from
|
48 |
-
|
|
|
|
|
|
|
49 |
|
50 |
-
tokenizer =
|
51 |
-
model =
|
52 |
-
"cosimoiaia/Loquace-
|
53 |
load_in_8bit=True,
|
54 |
device_map="auto",
|
55 |
quantization_config=BitsAndBytesConfig(
|
|
|
44 |
|
45 |
|
46 |
```python
|
47 |
+
from transformers import (
|
48 |
+
AutoTokenizer,
|
49 |
+
AutoModelForCausalLM,
|
50 |
+
BitsAndBytesConfig
|
51 |
+
)
|
52 |
|
53 |
+
tokenizer = AutoTokenizer.from_pretrained("cosimoiaia/Loquace-70m", padding_side="right", use_fast=True)
|
54 |
+
model = AutoModelForCausalLM.from_pretrained(
|
55 |
+
"cosimoiaia/Loquace-70m",
|
56 |
load_in_8bit=True,
|
57 |
device_map="auto",
|
58 |
quantization_config=BitsAndBytesConfig(
|