format code

ddPn08 · ddPn08 · commit 1175720393ac · 2023-05-23T09:48:01.000+09:00
diff --git a/lib/rvc/commons.py b/lib/rvc/commons.py
@@ -134,8 +134,6 @@ def generate_path(duration, mask):
     duration: [b, 1, t_x]
     mask: [b, 1, t_y, t_x]
     """
-    device = duration.device
-
     b, _, t_y, t_x = mask.shape
     cum_duration = torch.cumsum(duration, -1)
 
diff --git a/lib/rvc/models.py b/lib/rvc/models.py
@@ -848,4 +848,4 @@ def forward(self, y, y_hat):
             fmap_rs.append(fmap_r)
             fmap_gs.append(fmap_g)
 
-        return y_d_rs, y_d_gs, fmap_rs, fmap_gs
+        return y_d_rs, y_d_gs, fmap_rs, fmap_gs
diff --git a/lib/rvc/train.py b/lib/rvc/train.py
@@ -21,13 +21,20 @@
 from . import commons, utils
 from .checkpoints import save
 from .config import DatasetMetadata, TrainConfig
-from .data_utils import (DistributedBucketSampler, TextAudioCollate,
-                         TextAudioCollateMultiNSFsid, TextAudioLoader,
-                         TextAudioLoaderMultiNSFsid)
+from .data_utils import (
+    DistributedBucketSampler,
+    TextAudioCollate,
+    TextAudioCollateMultiNSFsid,
+    TextAudioLoader,
+    TextAudioLoaderMultiNSFsid,
+)
 from .losses import discriminator_loss, feature_loss, generator_loss, kl_loss
 from .mel_processing import mel_spectrogram_torch, spec_to_mel_torch
-from .models import (MultiPeriodDiscriminator, SynthesizerTrnMs256NSFSid,
-                     SynthesizerTrnMs256NSFSidNono)
+from .models import (
+    MultiPeriodDiscriminator,
+    SynthesizerTrnMs256NSFSid,
+    SynthesizerTrnMs256NSFSidNono,
+)
 
 
 def is_audio_file(file: str):
diff --git a/modules/core.py b/modules/core.py
@@ -38,7 +38,7 @@ def hash_check(url: str, out: str):
         hash = calc_sha256(out)
         return etag == hash
 
-    os.makedirs(os.path.join(MODELS_DIR, "pretrained", 'v2'), exist_ok=True)
+    os.makedirs(os.path.join(MODELS_DIR, "pretrained", "v2"), exist_ok=True)
 
     tasks = []
     for template in [
@@ -49,7 +49,7 @@ def hash_check(url: str, out: str):
     ]:
         basename = template.format("40")
         url = f"https://huggingface.co/ddPn08/rvc-webui-models/resolve/main/pretrained/v2/{basename}.pth"
-        out = os.path.join(MODELS_DIR, "pretrained", 'v2', f"{basename}.pth")
+        out = os.path.join(MODELS_DIR, "pretrained", "v2", f"{basename}.pth")
 
         if hash_check(url, out):
             continue
