Added initial support for Hidream E1 training

extensions_built_in/diffusion_models/__init__.py

@@ -1,10 +1,15 @@
 from .chroma import ChromaModel
-from .hidream import HidreamModel
+from .hidream import HidreamModel, HidreamE1Model
 from .f_light import FLiteModel
 from .omnigen2 import OmniGen2Model
 from .flux_kontext import FluxKontextModel
 
 AI_TOOLKIT_MODELS = [
     # put a list of models here
-    ChromaModel, HidreamModel, FLiteModel, OmniGen2Model, FluxKontextModel
+    ChromaModel, 
+    HidreamModel, 
+    HidreamE1Model, 
+    FLiteModel, 
+    OmniGen2Model, 
+    FluxKontextModel
 ]