|
|
--- |
|
|
license: mit |
|
|
--- |
|
|
|
|
|
## Model Card: UnfilteredAI/Promt-generator |
|
|
|
|
|
### Model Overview |
|
|
The **UnfilteredAI/Promt-generator** is a text generation model designed specifically for creating prompts for text-to-image models. It leverages **PyTorch** and **safetensors** for optimized performance and storage, ensuring that it can be easily deployed and scaled for prompt generation tasks. |
|
|
|
|
|
|
|
|
### Intended Use |
|
|
This model is primarily intended for: |
|
|
- **Prompt generation** for text-to-image models. |
|
|
- Creative AI applications where generating high-quality, diverse image descriptions is critical. |
|
|
- Supporting AI artists and developers working on generative art projects. |
|
|
|
|
|
### How to Use |
|
|
To generate prompts using this model, follow these steps: |
|
|
|
|
|
1. Load the model in your PyTorch environment. |
|
|
2. Input your desired parameters for the prompt generation task. |
|
|
3. The model will return text descriptions based on the input, which can then be used with text-to-image models. |
|
|
|
|
|
**Example Code:** |
|
|
|
|
|
```python |
|
|
from transformers import AutoModelForCausalLM, AutoTokenizer |
|
|
|
|
|
tokenizer = AutoTokenizer.from_pretrained("UnfilteredAI/Promt-generator") |
|
|
model = AutoModelForCausalLM.from_pretrained("UnfilteredAI/Promt-generator") |
|
|
|
|
|
prompt = "a red car" |
|
|
inputs = tokenizer(prompt, return_tensors="pt") |
|
|
outputs = model.generate(**inputs) |
|
|
generated_prompt = tokenizer.decode(outputs[0], skip_special_tokens=True) |
|
|
|
|
|
print(generated_prompt) |
|
|
``` |
