cosimoiaia
commited on
Commit
·
ff8d509
1
Parent(s):
79ea4d2
Update README.md
Browse files
README.md
CHANGED
@@ -50,9 +50,9 @@ from transformers import (
|
|
50 |
BitsAndBytesConfig
|
51 |
)
|
52 |
|
53 |
-
tokenizer = AutoTokenizer.from_pretrained("cosimoiaia/Loquace-
|
54 |
model = AutoModelForCausalLM.from_pretrained(
|
55 |
-
"cosimoiaia/Loquace-
|
56 |
load_in_8bit=True,
|
57 |
device_map="auto",
|
58 |
quantization_config=BitsAndBytesConfig(
|
|
|
50 |
BitsAndBytesConfig
|
51 |
)
|
52 |
|
53 |
+
tokenizer = AutoTokenizer.from_pretrained("cosimoiaia/Loquace-410m", padding_side="right", use_fast=True)
|
54 |
model = AutoModelForCausalLM.from_pretrained(
|
55 |
+
"cosimoiaia/Loquace-410m",
|
56 |
load_in_8bit=True,
|
57 |
device_map="auto",
|
58 |
quantization_config=BitsAndBytesConfig(
|