nlpulse
/

gpt-j-6b-english_quotes

Text Generation

Inference Endpoints

Model card Files Files and versions Community

egon-nlpulse commited on Jul 11, 2023

Commit

4a8eced

·

1 Parent(s): 8ac3a2d

ajustes

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -51,7 +51,7 @@ device = "cuda"
 text_list = ["Ask not what your country", "Be the change that", "You only live once, but", "I'm selfish, impatient and"]
 for text in text_list:
     inputs = tokenizer(text, return_tensors="pt").to(device)
-    outputs = model.generate(**inputs, max_new_tokens=20)
     print('>> ', text, " => ", tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```

 text_list = ["Ask not what your country", "Be the change that", "You only live once, but", "I'm selfish, impatient and"]
 for text in text_list:
     inputs = tokenizer(text, return_tensors="pt").to(device)
+    outputs = model.generate(**inputs, max_new_tokens=60)
     print('>> ', text, " => ", tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```