Add support for training Z-Image Turbo with a de-distill training adapter

extensions_built_in/diffusion_models/__init__.py

@@ -6,6 +6,7 @@ from .flux_kontext import FluxKontextModel
 from .wan22 import Wan225bModel, Wan2214bModel, Wan2214bI2VModel
 from .qwen_image import QwenImageModel, QwenImageEditModel, QwenImageEditPlusModel
 from .flux2 import Flux2Model
+from .z_image import ZImageModel
 
 AI_TOOLKIT_MODELS = [
     # put a list of models here
@@ -23,4 +24,5 @@ AI_TOOLKIT_MODELS = [
     QwenImageEditModel,
     QwenImageEditPlusModel,
     Flux2Model,
+    ZImageModel,
 ]