Add support for Baidu's ERNIE-Image (#793)

* Add support for ERNIE Image

* change float64 to float32

* Version bump

* Update ERNIE defaults
This commit is contained in:
Jaret Burkett (Ostris)
2026-04-14 09:45:12 -06:00
committed by GitHub
parent e868fca562
commit 3e0c904054
7 changed files with 834 additions and 4 deletions

View File

@@ -9,6 +9,7 @@ from .flux2 import Flux2Model, Flux2Klein4BModel, Flux2Klein9BModel
from .z_image import ZImageModel
from .ltx2 import LTX2Model, LTX23Model
from .zeta_chroma import ZetaChromaModel
from .ernie_image import ErnieImageModel
AI_TOOLKIT_MODELS = [
# put a list of models here
@@ -32,4 +33,5 @@ AI_TOOLKIT_MODELS = [
Flux2Klein4BModel,
Flux2Klein9BModel,
ZetaChromaModel,
ErnieImageModel,
]