1berry/fish-speech
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

"Ensure the version of checkpoints" (#752)

Browse Source
This commit is contained in:
PoTaTo 2024-12-15 12:46:17 +08:00 committed by GitHub
parent 3147dd26ef
commit f54c50fe62
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
API_FLAGS.txt
View File

@ -1,6 +1,6 @@
# --infer
--api
--listen 0.0.0.0:8080 \
--llama-checkpoint-path "checkpoints/fish-speech-1.4" \
--decoder-checkpoint-path "checkpoints/fish-speech-1.4/firefly-gan-vq-fsq-8x1024-21hz-generator.pth" \
--llama-checkpoint-path "checkpoints/fish-speech-1.5" \
--decoder-checkpoint-path "checkpoints/fish-speech-1.5/firefly-gan-vq-fsq-8x1024-21hz-generator.pth" \
--decoder-config-name firefly_gan_vq

2
dockerfile
View File

@ -1,7 +1,7 @@
FROM python:3.12-slim-bookworm AS stage-1
ARG TARGETARCH
ARG HUGGINGFACE_MODEL=fish-speech-1.4
ARG HUGGINGFACE_MODEL=fish-speech-1.5
ARG HF_ENDPOINT=https://huggingface.co
WORKDIR /opt/fish-speech

2
docs/pt/inference.md
View File

@ -70,7 +70,7 @@ Fornecemos uma API HTTP para inferência. O seguinte comando pode ser usado para
python -m tools.api_server \
--listen 0.0.0.0:8080 \
--llama-checkpoint-path "checkpoints/fish-speech-1.5" \
--decoder-checkpoint-path "checkpoints/fish-speech-1.4/firefly-gan-vq-fsq-8x1024-21hz-generator.pth" \
--decoder-checkpoint-path "checkpoints/fish-speech-1.5/firefly-gan-vq-fsq-8x1024-21hz-generator.pth" \
--decoder-config-name firefly_gan_vq
```

2
tools/llama/generate.py
View File

@ -1019,7 +1019,7 @@ def launch_thread_safe_queue_agent(
@click.option(
"--checkpoint-path",
type=click.Path(path_type=Path, exists=True),
default="checkpoints/fish-speech-1.4",
default="checkpoints/fish-speech-1.5",
)
@click.option("--device", type=str, default="cuda")
@click.option("--compile/--no-compile", default=False)

4
tools/vqgan/extract_vq.py
View File

@ -48,7 +48,7 @@ logger.add(sys.stderr, format=logger_format)
@lru_cache(maxsize=1)
def get_model(
config_name: str = "firefly_gan_vq",
checkpoint_path: str = "checkpoints/fish-speech-1.4/firefly-gan-vq-fsq-8x1024-21hz-generator.pth",
checkpoint_path: str = "checkpoints/fish-speech-1.5/firefly-gan-vq-fsq-8x1024-21hz-generator.pth",
device: str | torch.device = "cuda",
):
with initialize(version_base="1.3", config_path="../../fish_speech/configs"):
@ -139,7 +139,7 @@ def process_batch(files: list[Path], model) -> float:
@click.option("--config-name", default="firefly_gan_vq")
@click.option(
"--checkpoint-path",
default="checkpoints/fish-speech-1.4/firefly-gan-vq-fsq-8x1024-21hz-generator.pth",
default="checkpoints/fish-speech-1.5/firefly-gan-vq-fsq-8x1024-21hz-generator.pth",
)
@click.option("--batch-size", default=64)
@click.option("--filelist", default=None, type=Path)

2
tools/vqgan/inference.py
View File

@ -58,7 +58,7 @@ def load_model(config_name, checkpoint_path, device="cuda"):
@click.option("--config-name", default="firefly_gan_vq")
@click.option(
"--checkpoint-path",
default="checkpoints/fish-speech-1.4/firefly-gan-vq-fsq-8x1024-21hz-generator.pth",
default="checkpoints/fish-speech-1.5/firefly-gan-vq-fsq-8x1024-21hz-generator.pth",
)
@click.option(
"--device",