Add MOSS-Image SFT edit training

testing/test_model_loading.py

@@ -237,6 +237,20 @@ MODEL_TESTS = {
         },
         "sample": {**IMG, "num_inference_steps": 20, "guidance_scale": 3.0},
     },
+    "moss_image:edit": {
+        # edit LoRA path: vision-tower prompt + t=0 reference latents, isolated
+        # references with their K/V cached across sampling steps
+        "model": {
+            "name_or_path": "OpenMOSS-Team/MOSS-Image-SFT",
+            "quantize": True,
+            "quantize_te": True,
+            "qtype": "convrot8",
+            "qtype_te": "convrot8",
+            "model_kwargs": {"edit": True, "kv_cache": True, "match_target_res": True},
+        },
+        "sample": {**IMG, "num_inference_steps": 20, "guidance_scale": 3.0},
+        "needs_control_image": True,
+    },
     "moss_image:base_edit": {
         # edit LoRA path: vision-tower prompt + t=0 reference latents, isolated
         # references with their K/V cached across sampling steps