From b784ef11aecc358e22c16f63a23cc1322fb5d919 Mon Sep 17 00:00:00 2001 From: rx28 <66052512+rx28@users.noreply.github.com> Date: Sun, 17 Mar 2024 23:49:46 -0400 Subject: [PATCH] Update speech_dlm_criterion.py fix imports of metrics.py --- fairseq/criterions/speech_dlm_criterion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/fairseq/criterions/speech_dlm_criterion.py b/fairseq/criterions/speech_dlm_criterion.py index 8888180114..c01ef7ddcf 100644 --- a/fairseq/criterions/speech_dlm_criterion.py +++ b/fairseq/criterions/speech_dlm_criterion.py @@ -8,7 +8,8 @@ from typing import Optional import torch.nn.functional as F -from fairseq import metrics, utils +from fairseq import utils +from fairseq.logging import metrics from fairseq.criterions import FairseqCriterion, register_criterion from fairseq.dataclass import FairseqDataclass from omegaconf import II