add SetBatchSize for model

雾聪

2024-03-29 3ac539a58ca56a04123771ef145430cbcbdc294e

 funasr/models/transformer/search.py

@@ -9,7 +9,7 @@

import torch

from funasr.metrics import end_detect
from funasr.metrics.common import end_detect
from funasr.models.transformer.scorers.scorer_interface import PartialScorerInterface
from funasr.models.transformer.scorers.scorer_interface import ScorerInterface