unidflow-base-7B / README.md
onkarsus13's picture
Add files using upload-large-folder tool
0f20336 verified
|
Raw
History Blame Contribute Delete
849 Bytes
metadata
library_name: transformers
license: apache-2.0
pipeline_tag: image-to-image
tags:
  - multimodal
  - image-generation
  - image-to-image
  - masked-diffusion

Refined UniDFlow

This repository contains the released UniDFlow weights, tokenizer, and VQ-VAE prepared for the modular unidflow Python package.

from unidflow import UniDFlowPipeline

pipe = UniDFlowPipeline.from_pretrained("onkarsus13/unidflow-base-7B")
result = pipe(
    "image-to-image",
    image="input.jpg",
    prompt="Turn this photograph into a watercolor painting",
    edit_type="edit",
    seed=42,
)
result.save("edited.png")

UniDFlow is a bidirectional masked-diffusion model. Use UniDFlowPipeline rather than the autoregressive transformers.generate() method. Preserve the upstream UniDFlow citation and notices when redistributing this checkpoint.