update some deault config value (#19)

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

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

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]

tools/vqgan/inference.py

@@ -33,9 +33,7 @@ OmegaConf.register_new_resolver("eval", eval)
     "--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)