add

liuwenran · Aug 21, 2023 · 196032c · 196032c
1 parent 0fb4a67
commit 196032c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mmagic/structures/data_sample.py b/mmagic/structures/data_sample.py
@@ -244,6 +244,11 @@ def set_gt_label(
             self.gt_label = label
         return self
 
+    def set_gt_prompt(self, value: Union[str, Sequence[str]]) -> 'DataSample':
+        """Set label of ``gt_label``."""
+        self.prompt = value
+        return self
+
     @property
     def gt_label(self):
         """This the function to fetch gt label.