onediffusion-bf16 / README.md
twodgirl's picture
Update README.md
b75f2c5 verified
|
raw
history blame
637 Bytes
---
base_model:
- lehduong/OneDiffusion
license: cc-by-nc-4.0
---
# OneDiffusion
The weights are reuploaded in bfloat16, safetensors format. The T5 encoder output is not compatible with Flux/SD3, it needs to be redownloaded as well.
## Gradio
In the [instruction](https://github.com/lehduong/OneDiffusion?tab=readme-ov-file#quick-start) or in the [gradio demo](https://github.com/lehduong/OneDiffusion/blob/17ab6fb44dabf1ab43bbe053a79fe45fc307628d/gradio_demo.py#L132) change the repo_id to this url.
```python
pipeline = OneDiffusionPipeline.from_pretrained('twodgirl/onediffusion-bf16').to(device=device, dtype=torch.bfloat16)
```