optimum-intel-internal-testing/tiny-stable-diffusion-with-textual-inversion

optimum-intel-internal-testing

Texto a imagen

Pesos de adaptación de textual inversion para el modelo base `hf-internal-testing/tiny-stable-diffusion-torch`, orientados a generación de imágenes a partir de texto con Diffusers y StableDiffusionPipeline. El modelo corresponde a un ajuste fino text2image asociado a `katuni4ka/textual_inversion_cat`.

Como usar

Instalación y uso con Diffusers:
pip install -U diffusers transformers accelerate

import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained(
    "hf-internal-testing/tiny-stable-diffusion-torch",
    dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_textual_inversion(
    "optimum-intel-internal-testing/tiny-stable-diffusion-with-textual-inversion"
)

La tarjeta del modelo también incluye un marcador pendiente para añadir un ejemplo completo de ejecución del pipeline.

Funcionalidades

Tarea: generación de imágenes a partir de texto.
Formato y ecosistema: Diffusers, Safetensors y TensorBoard.
Pipeline compatible: `StableDiffusionPipeline` / `DiffusionPipeline`.
Tipo de adaptación: textual inversion sobre el modelo base `hf-internal-testing/tiny-stable-diffusion-torch`.
Licencia: `creativeml-openrail-m`.
No está desplegado en proveedores de inferencia de Hugging Face según la página.

Casos de uso

Probar cargas de textual inversion en pipelines de Stable Diffusion pequeños.
Validar integración con Diffusers usando un modelo interno de prueba de Hugging Face.
Experimentar con adaptación text2image mediante pesos de textual inversion.
Ejecutar pruebas ligeras de compatibilidad con CUDA o MPS en flujos Diffusers.