Update model_config.json

model_config.json

@@ -1,32 +1,34 @@
 {
   "_class_name": "flux_dev_finetune",
   "_diffusers_version": "0.30.0.dev0",
-  "scheduler": [
-    "diffusers",
-    "FlowMatchEulerDiscreteScheduler"
-  ],
-  "text_encoder": [
-    "transformers",
-    "CLIPTextModel"
-  ],
-  "text_encoder_2": [
-    "transformers",
-    "T5EncoderModel"
-  ],
-  "tokenizer": [
-    "transformers",
-    "CLIPTokenizer"
-  ],
-  "tokenizer_2": [
-    "transformers",
-    "T5TokenizerFast"
-  ],
-  "transformer": [
-    "diffusers",
-    "FluxTransformer2DModel"
-  ],
-  "vae": [
-    "diffusers",
-    "AutoencoderKL"
-  ]
+  "scheduler": {
+    "type": "FlowMatchEulerDiscreteScheduler",
+    "library": "diffusers"
+  },
+  "text_encoder": {
+    "type": "CLIPTextModel",
+    "library": "transformers"
+  },
+  "text_encoder_2": {
+    "type": "T5EncoderModel",
+    "library": "transformers"
+  },
+  "tokenizer": {
+    "type": "CLIPTokenizer",
+    "library": "transformers"
+  },
+  "tokenizer_2": {
+    "type": "T5TokenizerFast",
+    "library": "transformers"
+  },
+  "transformer": {
+    "type": "FluxTransformer2DModel",
+    "library": "diffusers"
+  },
+  "vae": {
+    "type": "AutoencoderKL",
+    "library": "diffusers"
+  },
+  "model_type": "diffusion",  // Specify model type for clarity
+  "pipeline_tag": "text-to-image"  // Indicate the task the model is meant for
 }