How to use from the
Use from the
Diffusers library
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("jieeliu/EraserDiT", dtype=torch.bfloat16, device_map="cuda")

prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]

EraserDiT: Fast Video Inpainting with Diffusion Transformer Model

Huggingface Model Github arXiv Demo Page


EraserDiT Model Card

This model card focuses on the model associated with the EraserDiT model.

Downloads last month
33
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for jieeliu/EraserDiT