Update mellon_pipeline_config.json

mellon_pipeline_config.json

@@ -17,53 +17,82 @@
           "label": "Quant Type",
           "type": "string",
           "options": [
-            "bnb_4bit"
+            "bnb_4bit",
+            "bnb_8bit"
           ],
-          "value": "bnb_4bit"
+          "value": "bnb_4bit",
+          "onChange": {
+            "bnb_4bit": [
+              "bnb_4bit_quant_type",
+              "bnb_4bit_compute_dtype",
+              "bnb_4bit_use_double_quant"
+            ],
+            "bnb_8bit": [
+              "llm_int8_threshold",
+              "llm_int8_has_fp16_weight"
+            ]
+          }
         },
         "bnb_4bit_quant_type": {
-          "label": "Bnb 4Bit Quant Type",
+          "label": "4-bit Quant Type",
           "type": "string",
           "options": [
-            "nf4"
+            "nf4",
+            "fp4"
           ],
           "value": "nf4"
         },
         "bnb_4bit_compute_dtype": {
-          "label": "Bnb 4Bit Compute Dtype",
+          "label": "Compute Dtype",
           "type": "string",
           "options": [
-            ""
+            "",
+            "float32",
+            "float16",
+            "bfloat16"
           ],
           "value": ""
         },
         "bnb_4bit_use_double_quant": {
-          "label": "Bnb 4Bit Use Double Quant",
+          "label": "Double Quant",
           "type": "boolean",
           "default": false
         },
         "llm_int8_threshold": {
-          "label": "Llm Int8 Threshold",
+          "label": "Int8 Threshold",
           "type": "float",
           "display": "slider",
           "default": 6.0,
-          "min": 6.0,
-          "max": 6.0,
-          "step": 0.01
+          "min": 0.0,
+          "max": 10.0,
+          "step": 0.5
         },
         "llm_int8_has_fp16_weight": {
-          "label": "Llm Int8 Has Fp16 Weight",
+          "label": "Has FP16 Weight",
           "type": "boolean",
           "default": false
         },
         "llm_int8_skip_modules": {
-          "label": "Llm Int8 Skip Modules",
-          "type": "custom",
-          "display": "input"
+          "label": "Skip Modules",
+          "type": "string",
+          "display": "select",
+          "options": [
+            "transformer_blocks.0.img_mod",
+            "transformer_blocks.33.img_mod",
+            "transformer_blocks.58.attn.to_k",
+            "transformer_blocks.59.attn.to_out",
+            "time_text_embed",
+            "img_in",
+            "txt_in"
+          ],
+          "fieldOptions": {
+            "multiple": true
+          },
+          "value": []
         },
         "quantization_config": {
           "label": "Quantization Config",
-          "type": "custom",
+          "type": "quant_config",
           "display": "output"
         },
         "doc": {