From 8da57a04496ee10d3e3630799a5def95281500a1 Mon Sep 17 00:00:00 2001 From: JinZr Date: Fri, 6 Sep 2024 21:21:58 +0800 Subject: [PATCH] black formatted --- egs/libritts/CODEC/encodec/encodec.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/egs/libritts/CODEC/encodec/encodec.py b/egs/libritts/CODEC/encodec/encodec.py index bde03034f..385551d06 100644 --- a/egs/libritts/CODEC/encodec/encodec.py +++ b/egs/libritts/CODEC/encodec/encodec.py @@ -75,7 +75,8 @@ class Encodec(nn.Module): reuse_cache = False e = self.encoder(speech) index = torch.tensor( - random.randint(0, len(self.target_bandwidths) - 1), device=speech.device, + random.randint(0, len(self.target_bandwidths) - 1), + device=speech.device, ) if torch.distributed.is_initialized(): torch.distributed.broadcast(index, src=0) @@ -173,7 +174,8 @@ class Encodec(nn.Module): reuse_cache = False e = self.encoder(speech) index = torch.tensor( - random.randint(0, len(self.target_bandwidths) - 1), device=speech.device, + random.randint(0, len(self.target_bandwidths) - 1), + device=speech.device, ) if torch.distributed.is_initialized(): torch.distributed.broadcast(index, src=0)