diff --git a/api.py b/api.py index dd4da68..be872c3 100644 --- a/api.py +++ b/api.py @@ -59,6 +59,7 @@ model_names = { "ErnieImage": "ernie-image", "Flux2": "flux2", "Anima": "anima", + "Krea2": "krea2", } gguf_architectures = {