From 9c5f00794e7c474d1079ff78e5c69e8c53a30e6e Mon Sep 17 00:00:00 2001
From: dudulu <solacowa@gmail.com>
Date: 星期一, 05 二月 2024 10:10:52 +0800
Subject: [PATCH] 修复chunk_size可能传入str或list结构 (#1357)

---
 funasr/losses/label_smoothing_loss.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/losses/label_smoothing_loss.py b/funasr/losses/label_smoothing_loss.py
index c272ea8..8f0809a 100644
--- a/funasr/losses/label_smoothing_loss.py
+++ b/funasr/losses/label_smoothing_loss.py
@@ -8,7 +8,7 @@
 
 import torch
 from torch import nn
-from funasr.modules.nets_utils import make_pad_mask
+from funasr.models.transformer.utils.nets_utils import make_pad_mask
 
 
 class LabelSmoothingLoss(nn.Module):

--
Gitblit v1.9.1