OneFlow diffusers

OneFlow backend support for diffusers

Getting started

Please refer to this wiki

Quick demo

python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.15.1"
python3 -m pip uninstall accelerate -y
python3 -m pip install -U onediff
python3 -m onediff.demo

More about OneFlow

OneFlow's main repo

Development

Option 1: Fresh clone and dev install

git clone https://github.com/Oneflow-Inc/diffusers.git onediff
cd onediff
python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.15.1"
python3 -m pip uninstall accelerate -y
python3 -m pip install -e .

Option 2: Setup if you were using the the oneflow-fork branch before

1. uninstall transformers and diffusers

python3 -m pip uninstall transformers -y
python3 -m pip uninstall diffusers -y

2. install transformers and diffusers

python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.15.1"
python3 -m pip uninstall accelerate -y

3. delete the main first:

git branch -D main
git fetch
git checkout main
python3 -m pip install -e .

More examples

There is a directory for examples

Release

• run examples to check it works

  python3 examples/text_to_image.py
  python3 examples/text_to_image_dpmsolver.py

• bump version in these files:

  setup.py
  src/onediff/__init__.py
  

• build wheel

  rm -rf dist
  python3 setup.py bdist_wheel
  

• upload to pypi

  twine upload dist/*