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:
committed by
GitHub
parent
e868fca562
commit
3e0c904054
@@ -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,
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user