Update README.md
Browse files
README.md
CHANGED
@@ -34,7 +34,10 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
|
|
34 |
|
35 |
# Load the tokenizer and model
|
36 |
tokenizer = AutoTokenizer.from_pretrained("webpolis/zenos-gpt-j-6B-instruct-4bit")
|
37 |
-
model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
|
|
|
38 |
|
39 |
user_msg = '''Escribe un poema breve utilizando los siguientes conceptos:
|
40 |
|
|
|
34 |
|
35 |
# Load the tokenizer and model
|
36 |
tokenizer = AutoTokenizer.from_pretrained("webpolis/zenos-gpt-j-6B-instruct-4bit")
|
37 |
+
model = AutoModelForCausalLM.from_pretrained(
|
38 |
+
"webpolis/zenos-gpt-j-6B-instruct-4bit",
|
39 |
+
use_safetensors=True
|
40 |
+
)
|
41 |
|
42 |
user_msg = '''Escribe un poema breve utilizando los siguientes conceptos:
|
43 |
|