refactor: clean up num_examples code (#14)

- refactor: clean up num_examples code (b67df152d4d5143d84ff7c023a245d301ecccb22)

custom_st.py

@@ -104,11 +104,7 @@ class Transformer(nn.Module):
         adapter_mask = None
         if task_type:
             task_id = self._adaptation_map[task_type]
-            num_examples = 1
-            if isinstance(features['input_ids'][0], list):
-                # If input_ids[0] is a list, it means multiple inputs (list of texts)
-                num_examples = len(features['input_ids'])
-
+            num_examples = features['input_ids'].size(0)
             adapter_mask = torch.full(
                 (num_examples,), task_id, dtype=torch.int32, device=features['input_ids'].device
             )