fix

sampling.py

@@ -49,11 +49,10 @@ class CAIFSampler:
         top_k_classifier,
         classifier_weight,
         caif_tokens_num=None,
-        act_type: str = "softmax",
+        act_type: str = "sigmoid",
+        target_cls_id: int = 0,
         **kwargs
     ):
-        target_cls_id = kwargs["target_cls_id"]
-        act_type = kwargs["act_type"]
         next_token_logits = output_logis[:, -1]
         next_token_log_probs = F.log_softmax(
             next_token_logits, dim=-1