webpolis commited on
Commit
5e0c2b6
1 Parent(s): c8a70ff

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
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("webpolis/zenos-gpt-j-6B-instruct-4bit", use_safetensors=True)
 
 
 
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