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
30
Safetensors
Model size
2B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for jieeliu/EraserDiT