Quantization made by Richard Erkhov.
Promt-generator - bnb 8bits
- Model creator: https://huggingface.co/UnfilteredAI/
- Original model: https://huggingface.co/UnfilteredAI/Promt-generator/
Original model description:
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:
- Load the model in your PyTorch environment.
- Input your desired parameters for the prompt generation task.
- The model will return text descriptions based on the input, which can then be used with text-to-image models.
Example Code:
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)