OneFlow backend support for diffusers
Please refer to this wiki
python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.19.3"
python3 -m pip uninstall accelerate -y
python3 -m pip install -U onediff
python3 -m onediff.demo
OneFlow's main repo
git clone https://github.com/Oneflow-Inc/diffusers.git onediff
cd onediff
python3 -m pip install -e .
python3 -m pip install -r examples/requirements.txt
python3 -m pip uninstall accelerate -y
python3 examples/text_to_image.py
-
run examples to check it works
python3 examples/text_to_image.py --model_id=... python3 examples/text_to_image_sdxl.py --base ... bash examples/unet_save_and_load.sh --model_id=...
-
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/*