mirror/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2026-04-24 03:59:32 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(diffusers): support flux models (#3129)

Browse Source 
* feat(diffusers): support flux models

This adds support for FLUX models. For instance:
https://huggingface.co/black-forest-labs/FLUX.1-dev

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

* feat(diffusers): support FluxTransformer2DModel

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

* Small fixups

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

---------

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto
2024-08-11 01:31:53 +02:00
committed by GitHub
parent 7ba4a78fcc
commit 74eaf02484
6 changed files with 45 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/python/diffusers/requirements-cublas11.txt
+2 -1
View File
@@ -6,4 +6,5 @@ transformers
accelerate
compel
peft
sentencepiece
sentencepiece
optimum-quanto