Add support to train MOSS Image Base as an edit model

testing/test_model_loading.py

@@ -237,6 +237,20 @@ MODEL_TESTS = {
         },
         "sample": {**IMG, "num_inference_steps": 20, "guidance_scale": 3.0},
     },
+    "moss_image:base_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-Base",
+            "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:comfy": {
         # the Comfy-Org repack files named outright (a bare repo id prefers any
         # local bf16): int8 convrot SFT DiT + Qwen3.5-9B TE attach as-is