update some deault config value (#19)

fishaudio · Dec 20, 2023 · fa62856 · fa62856
1 parent 744622d
commit fa62856
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 6 deletions.
diff --git a/fish_speech/configs/vqgan_finetune.yaml b/fish_speech/configs/vqgan_finetune.yaml
@@ -9,7 +9,7 @@ resume_weights_only: true
 # Lightning Trainer
 trainer:
   accelerator: gpu
-  devices: 4
+  devices: auto
   strategy: ddp_find_unused_parameters_true
   precision: 32
   max_steps: 100000

diff --git a/fish_speech/configs/vqgan_pretrain.yaml b/fish_speech/configs/vqgan_pretrain.yaml
@@ -7,7 +7,7 @@ project: vqgan
 # Lightning Trainer
 trainer:
   accelerator: gpu
-  devices: 4
+  devices: auto
   strategy: ddp_find_unused_parameters_true
   precision: 32
   max_steps: 1_000_000

diff --git a/pyproject.toml b/pyproject.toml
@@ -35,7 +35,8 @@ dependencies = [
     "tensorboard",
     "grpcio>=1.58.0",
     "kui>=1.6.0",
-    "zibai-server>=0.9.0"
+    "zibai-server>=0.9.0",
+    "loguru",
 ]
 
 [build-system]

diff --git a/tools/vqgan/inference.py b/tools/vqgan/inference.py
@@ -33,9 +33,7 @@
     "--output-path", "-o", default="fake.wav", type=click.Path(path_type=Path)
 )
 @click.option("--config-name", "-cfg", default="vqgan_pretrain")
-@click.option(
-    "--checkpoint-path", "-ckpt", default="checkpoints/vqgan/step_000380000_wo.ckpt"
-)
+@click.option("--checkpoint-path", "-ckpt", default="checkpoints/vqgan-v1.pth")
 def main(input_path, output_path, config_name, checkpoint_path):
     with initialize(version_base="1.3", config_path="../../fish_speech/configs"):
         cfg = compose(config_name=config_name)