gokaygokay
/

Flux-Prompt-Enhance

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

gokaygokay commited on Aug 27

Commit

f0b2c31

•

1 Parent(s): 9f38f29

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -26,6 +26,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 enhancer = pipeline('text2text-generation',
                     model=model,
                     tokenizer=tokenizer,
                     device=device)
 max_target_length = 256

 enhancer = pipeline('text2text-generation',
                     model=model,
                     tokenizer=tokenizer,
+                    repetition_penalty= 1.2,
                     device=device)
 max_target_length = 256