From 23f448aa93888699786d7446767794a0282c75ed Mon Sep 17 00:00:00 2001
From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Date: Mon, 27 Nov 2023 20:34:31 +0800
Subject: [PATCH] [Fix] RTMHead import error

---
 mmrazor/models/task_modules/demo_inputs/mmpose_demo_input.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmrazor/models/task_modules/demo_inputs/mmpose_demo_input.py b/mmrazor/models/task_modules/demo_inputs/mmpose_demo_input.py
index dbf5f2772..66f316018 100644
--- a/mmrazor/models/task_modules/demo_inputs/mmpose_demo_input.py
+++ b/mmrazor/models/task_modules/demo_inputs/mmpose_demo_input.py
@@ -29,7 +29,7 @@ def demo_mmpose_inputs(model, for_training=False, batch_size=1):
     imgs = torch.randn(*input_shape)
 
     batch_data_samples = []
-    from mmpose.models.heads import RTMHead
+    from mmpose.models.heads import RTMCCHead as RTMHead
     if isinstance(model.head, HeatmapHead):
         batch_data_samples = get_packed_inputs(
             batch_size,