fix(wan2.2): enforce native save structure and align default dual guidance

Author: Charles2530

configs/quantization/video_gen/wan2_2_t2v/awq_w_a.yaml

@@ -5,6 +5,9 @@ model:
     path: /mnt/lm_data_afs/wangzining/charles/lab/llmc/model/Wan2.2-T2V-A14B
     # 若未 `pip install -e /path/to/Wan2.2`，可显式指定官方仓库代码路径：
     # wan2_repo_path: /path/to/Wan2.2
+    # 默认严格走官方 Wan2.2 原生后端；官方代码不可用时会直接报错，不再静默回退到 Diffusers。
+    # 若确实需要回退可开启：
+    # allow_diffusers_fallback: True
     torch_dtype: auto
     # 显存不足时开启：校准阶段捕获的激活存到 CPU，量化时再按 block 搬到 GPU
     use_cpu_to_save_cuda_mem_for_catcher: True
@@ -17,7 +20,9 @@ calib:
     target_height: 480 # OOM 时可减小，如 320
     target_width: 832  # OOM 时可减小，如 576
     num_frames: 81    # OOM 时可减小，如 49 或 33
-    guidance_scale: 5.0
+    # 对齐官方 Wan2.2 默认 sample_guide_scale=(3.0, 4.0) (low_noise, high_noise)
+    guidance_scale: 4.0      # high_noise
+    guidance_scale_2: 3.0    # low_noise
     seed: *seed
 eval:
     eval_pos: [transformed, fake_quant]
@@ -29,7 +34,9 @@ eval:
     target_height: 480
     target_width: 832
     num_frames: 81
-    guidance_scale: 5.0
+    # 对齐官方 Wan2.2 默认 sample_guide_scale=(3.0, 4.0) (low_noise, high_noise)
+    guidance_scale: 4.0      # high_noise
+    guidance_scale_2: 3.0    # low_noise
     output_video_path: ./output_videos_awq/
 quant:
     video_gen:

configs/quantization/video_gen/wan2_2_t2v/awq_w_a_skip_first.yaml

@@ -5,6 +5,9 @@ model:
     path: /mnt/lm_data_afs/wangzining/charles/lab/llmc/model/Wan2.2-T2V-A14B
     # 若未 `pip install -e /path/to/Wan2.2`，可显式指定官方仓库代码路径：
     # wan2_repo_path: /path/to/Wan2.2
+    # 默认严格走官方 Wan2.2 原生后端；官方代码不可用时会直接报错，不再静默回退到 Diffusers。
+    # 若确实需要回退可开启：
+    # allow_diffusers_fallback: True
     torch_dtype: auto
     # 显存不足时开启：校准阶段捕获的激活存到 CPU，量化时再按 block 搬到 GPU
     use_cpu_to_save_cuda_mem_for_catcher: True
@@ -17,7 +20,9 @@ calib:
     target_height: 480 # OOM 时可减小，如 320
     target_width: 832  # OOM 时可减小，如 576
     num_frames: 81    # OOM 时可减小，如 49 或 33
-    guidance_scale: 5.0
+    # 对齐官方 Wan2.2 默认 sample_guide_scale=(3.0, 4.0) (low_noise, high_noise)
+    guidance_scale: 4.0      # high_noise
+    guidance_scale_2: 3.0    # low_noise
     seed: *seed
 eval:
     eval_pos: [transformed, fake_quant]
@@ -29,7 +34,9 @@ eval:
     target_height: 480
     target_width: 832
     num_frames: 81
-    guidance_scale: 5.0
+    # 对齐官方 Wan2.2 默认 sample_guide_scale=(3.0, 4.0) (low_noise, high_noise)
+    guidance_scale: 4.0      # high_noise
+    guidance_scale_2: 3.0    # low_noise
     output_video_path: ./output_videos_awq_skip_first/
 quant:
     video_gen:

llmc/compression/quantization/base_blockwise_quantization.py

@@ -1060,6 +1060,58 @@ def contiguous_params(self):
                     if not param.is_contiguous():
                         param.data = param.data.contiguous()
 
+    def _copy_wan22_native_checkpoint(self, src, dst):
+        if not isinstance(src, str) or not os.path.isdir(src):
+            raise RuntimeError(
+                'Wan2.2 official save expects a local native checkpoint directory, '
+                f'but got src={src!r}.'
+            )
+        if os.path.abspath(src) == os.path.abspath(dst):
+            raise RuntimeError(
+                'Wan2.2 official save path must differ from source checkpoint path '
+                f'(src=dst={src}).'
+            )
+        if os.path.exists(dst):
+            shutil.rmtree(dst)
+        shutil.copytree(src, dst)
+        logger.info(f'Copied original Wan2.2 native checkpoint from {src} to {dst}')
+
+    def _validate_wan22_native_save_structure(self, save_path, source_path=None):
+        if not os.path.isdir(save_path):
+            raise RuntimeError(f'Wan2.2 saved path is not a directory: {save_path}')
+
+        required_entries = ['configuration.json', 'high_noise_model', 'low_noise_model']
+        missing_required = [
+            name for name in required_entries
+            if not os.path.exists(os.path.join(save_path, name))
+        ]
+        if missing_required:
+            raise RuntimeError(
+                'Wan2.2 saved structure is incomplete. Missing required entries: '
+                f'{missing_required}. save_path={save_path}'
+            )
+
+        if isinstance(source_path, str) and os.path.isdir(source_path):
+            source_entries = set(os.listdir(source_path))
+            source_non_expert_entries = sorted(
+                name for name in source_entries
+                if name not in {'high_noise_model', 'low_noise_model'}
+            )
+            missing_non_expert = [
+                name for name in source_non_expert_entries
+                if not os.path.exists(os.path.join(save_path, name))
+            ]
+            if missing_non_expert:
+                raise RuntimeError(
+                    'Wan2.2 saved structure lost original non-expert files/directories: '
+                    f'{missing_non_expert}. source_path={source_path}, save_path={save_path}'
+                )
+
+        logger.info(
+            f'Wan2.2 native save structure verified. '
+            f'top-level entries={sorted(os.listdir(save_path))}'
+        )
+
     @torch.no_grad()
     def save_model(self, path):
         if int(os.environ['RANK']) != 0:
@@ -1082,16 +1134,8 @@ def save_model(self, path):
             self.copy_tokenizer(path)
         elif self.config.model.type in ['Wan2T2V']:
             if getattr(self.model.Pipeline, '_is_wan_official', False):
-                src = self.model.model_path
-                copied_from_source = False
-                if isinstance(src, str) and os.path.isdir(src) and os.path.abspath(src) != os.path.abspath(path):
-                    if os.path.exists(path):
-                        shutil.rmtree(path)
-                    shutil.copytree(src, path)
-                    logger.info(f'Copied original Wan2.2 native checkpoint from {src} to {path}')
-                    copied_from_source = True
-                if not copied_from_source and os.path.exists(path):
-                    shutil.rmtree(path)
+                src = getattr(self.model, 'pipeline_model_path', self.model.model_path)
+                self._copy_wan22_native_checkpoint(src, path)
 
                 self.model.Pipeline.transformer.save_pretrained(
                     os.path.join(path, 'high_noise_model')
@@ -1105,6 +1149,7 @@ def save_model(self, path):
                         os.path.join(path, 'low_noise_model')
                     )
                     logger.info('save Wan2.2 low_noise_model done --')
+                self._validate_wan22_native_save_structure(path, source_path=src)
                 return
 
             # Copy the full original pipeline (VAE, text encoder, tokenizer, scheduler, etc.)

llmc/models/wan2_2_t2v.py

@@ -163,6 +163,24 @@ def _has_wan22_native_layout(model_path):
             and os.path.isdir(os.path.join(model_path, 'low_noise_model'))
         )
 
+    @staticmethod
+    def _is_wan22_native_repo_id(model_path):
+        if not isinstance(model_path, str):
+            return False
+        return model_path.rstrip('/\\') == 'Wan-AI/Wan2.2-T2V-A14B'
+
+    def _should_require_official_backend(self, normalized_model_path):
+        if self.config.model.get('force_diffusers', False):
+            return False
+        if self.config.model.get('diffusers_path', None):
+            return False
+        if self.config.model.get('allow_diffusers_fallback', False):
+            return False
+        return (
+            self._has_wan22_native_layout(normalized_model_path)
+            or self._is_wan22_native_repo_id(normalized_model_path)
+        )
+
     def _import_official_wan(self):
         def _import_impl():
             from wan.configs import t2v_A14B
@@ -185,7 +203,8 @@ def _import_impl():
                     )
             logger.warning(
                 'Failed to import official Wan2.2 runtime (wan package). '
-                f'Falling back to diffusers import path. import_error={e}'
+                'Diffusers fallback depends on model.allow_diffusers_fallback/model.force_diffusers. '
+                f'import_error={e}'
             )
             return None, None
 
@@ -286,6 +305,9 @@ def _resolve_pipeline_model_path(self):
 
     def build_model(self):
         self.use_official_wan = False
+        normalized_model_path = self._normalize_hf_repo_path(self.model_path)
+        require_official_backend = self._should_require_official_backend(normalized_model_path)
+
         if self._try_build_official_wan_pipeline():
             self.find_llmc_model()
             self.find_blocks()
@@ -302,6 +324,16 @@ def build_model(self):
             logger.info('Model: %s', self.model)
             return
 
+        if require_official_backend:
+            raise RuntimeError(
+                'Detected Wan2.2 native source '
+                f'({normalized_model_path}) but official Wan runtime is unavailable. '
+                'Please install/prepare official Wan2.2 code (pip install -e /path/to/Wan2.2 '
+                'or set model.wan2_repo_path). '
+                'If you intentionally want Diffusers fallback, set '
+                'model.allow_diffusers_fallback=True or model.force_diffusers=True.'
+            )
+
         self.pipeline_model_path = self._resolve_pipeline_model_path()
         vae = AutoencoderKLWan.from_pretrained(
             self.pipeline_model_path,