|
--- |
|
base_model: google-t5/t5-base |
|
datasets: |
|
- gokaygokay/prompt-enhancer-dataset |
|
language: |
|
- en |
|
library_name: transformers |
|
license: apache-2.0 |
|
pipeline_tag: text2text-generation |
|
--- |
|
|
|
```python |
|
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM |
|
|
|
device = "cuda" if torch.cuda.is_available() else "cpu" |
|
|
|
# Model checkpoint |
|
model_checkpoint = "gokaygokay/Flux-Prompt-Enhance" |
|
|
|
# Tokenizer |
|
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint) |
|
|
|
# Model |
|
model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint) |
|
|
|
enhancer = pipeline('text2text-generation', |
|
model=model, |
|
tokenizer=tokenizer, |
|
repetition_penalty= 1.2, |
|
device=device) |
|
|
|
max_target_length = 256 |
|
prefix = "enhance prompt: " |
|
|
|
short_prompt = "beautiful house with text 'hello'" |
|
answer = enhancer(prefix + short_prompt, max_length=max_target_length) |
|
final_answer = answer[0]['generated_text'] |
|
print(final_answer) |
|
|
|
# a two-story house with white trim, large windows on the second floor, |
|
# three chimneys on the roof, green trees and shrubs in front of the house, |
|
# stone pathway leading to the front door, text on the house reads "hello" in all caps, |
|
# blue sky above, shadows cast by the trees, sunlight creating contrast on the house's facade, |
|
# some plants visible near the bottom right corner, overall warm and serene atmosphere. |
|
``` |
