How to use from the
Use from the
Transformers library
# Gated model: Login with a HF token with gated access permission
hf auth login
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("image-to-image", model="onkarsus13/unidflow-base-7B")
# Load model directly
from transformers import UniDFlowForMultiModalGeneration
model = UniDFlowForMultiModalGeneration.from_pretrained("onkarsus13/unidflow-base-7B", device_map="auto")
Quick Links

You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

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.

Downloads last month
3
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support