2) in representative tasks (e.g., text-to-image generation). You can find the original codebase at thu-ml/unidiffuser and additional checkpoints at thu-ml. There is currently an issue on PyTorch 1.X where the output images are all black or the pixel values become NaNs. This issue can be mitigated by switching to PyTorch 2.X. This pipeline was contributed by dg845. ❤️  Usage Examples Because the UniDiffuser model is trained to model the joint di