decoding

游雁

2024-06-13 e42c693f0b43648c65bc67908bdb2824cb24ff80

 funasr/models/llm_asr/model.py

@@ -420,6 +420,8 @@
                        layer_id = int(name[beg + 1 : end - 1])
                        if layer_id in freeze_layer_num:
                            param.requires_grad = False
                    else:
                        param.requires_grad = False
                else:
                    param.requires_grad = False

@@ -526,7 +528,7 @@
                ]

        labels_ids[labels_ids == -1] = -100

        attention_mask[attention_mask < 0] = 0
        model_outputs = self.llm(
            inputs_embeds=inputs_embeds, attention_mask=attention_mask, labels=labels_ids
        )