unidisc / README.md
aswerdlow's picture
Initial commit
131da64
|
raw
history blame
2.01 kB

Unified Multimodal Discrete Diffusion

Alexander Swerdlow1*  Mihir Prabhudesai1*  Siddharth Gandhi1  Deepak Pathak1  Katerina Fragkiadaki1 

1 Carnegie Mellon University 

ArXiv Webpage

Hugging Face models and annotations

The UniDisc checkpoints are available on Hugging Face:

Getting Started

To install the dependencies, run:

git submodule update --init --recursive
uv sync --no-group dev
uv sync

For a more detailed installation guide, please refer to INSTALL.md.

Training

See TRAIN.md for details.

Inference

Interactive demo for TODO.

python demo/server.py
python demo/client_simple_fasthtml.py

Training

See TRAINING.md for details.

Evaluation

See EVAL.md for details.

Citation

To cite our work, please use the following:

@article{TODO,
  title={TODO},
  author={TODO},
  journal={arXiv preprint arXiv:TODO},
  year={TODO}
}

Credits

This repository is built on top of the following repositories: