Added support for finetuning OmniGen2.

extensions_built_in/diffusion_models/__init__.py

@@ -1,8 +1,9 @@
 from .chroma import ChromaModel
 from .hidream import HidreamModel
 from .f_light import FLiteModel
+from .omnigen2 import OmniGen2Model
 
 AI_TOOLKIT_MODELS = [
     # put a list of models here
-    ChromaModel, HidreamModel, FLiteModel
+    ChromaModel, HidreamModel, FLiteModel, OmniGen2Model
 ]