| """Head-only voice-pack preset (unused by the CLI).""" |
| import argparse |
| import json |
| import urllib.error |
| from pathlib import Path |
|
|
| from vlib import ctx |
| from vlib.fetch import _fetch_pack, _resolve_output_name, _source_arch |
| from vlib.sources import (_family, _hf_config, _public_source, |
| _slugify_voice_name, open_source) |
| from vlib.tensors import write_safetensors |
| from vlib.ui import _fail, _note, _now_iso, _say, _step |
|
|
|
|
| def _resolve_pack_tensors(source, repo, config, names): |
| """Head-only voice-pack preset. Embed/norm/blocks stay manual |
| (voice get <repo> <tensors...>). |
| Config is the source config with _hf_config fallback. Miss fails loud.""" |
| if config is None and getattr(source, "repo", None): |
| config = _hf_config(source.repo) |
| head = _resolve_output_name(source, config, names) |
| if head is None: |
| _fail(f" ✗ pack: no output head found in '{repo}'.") |
| return [head] |
|
|
|
|
| def cmd_pack(args): |
| repo = args.repo.strip().rstrip("/") |
| if not repo: |
| _fail(" ✗ Enter a model name — for example: voice pack google/gemma-3-1b") |
| _step(f"Finding the voice pack in {repo}…") |
| try: |
| source = open_source(repo) |
| names = source.names() |
| if not names: |
| _fail(f" ✗ No tensors found in '{repo}'.") |
| arch, config = _source_arch(source) |
| except urllib.error.HTTPError as e: |
| _fail(f" ✗ Model '{repo}' was not found or is not accessible ({e.code}).") |
| except Exception as e: |
| if ctx.VERBOSE: |
| import traceback |
| _say(traceback.format_exc()) |
| _fail(f" ✗ Could not reach the model. Check your connection ({type(e).__name__}).") |
|
|
| _step("Resolving voice-pack tensors…") |
| resolved = _resolve_pack_tensors(source, repo, config, names) |
| for tname in resolved: |
| _note(f" {tname}") |
| |
| |
| pack = _fetch_pack(source, repo, resolved) |
| slug = _slugify_voice_name(repo.split("/")[-1].split("@")[0].split(":")[0].replace(".gguf", "").replace(".safetensors", "")) |
| name = f"pack-{slug}" |
| import tempfile, shutil |
| stage = Path(tempfile.mkdtemp(prefix="voicepack-stage-")) |
| try: |
| st_path = stage / "voice.safetensors" |
| write_safetensors(pack, str(st_path)) |
| js_path = stage / "voice.json" |
| js_path.write_text(json.dumps({ |
| "name": name, |
| "source": _public_source(repo), |
| "source_format": source.kind, |
| "tensors": [{"name": n, "shape": [int(x) for x in pack[n][2]], "dtype": pack[n][1]} |
| for n in resolved if n in pack], |
| "architecture": arch, |
| "family": _family(arch), |
| "downloaded_at": _now_iso(), |
| }, indent=2) + "\n") |
| from vlib.cmds_voices import cmd_add |
| cmd_add(argparse.Namespace(safetensors=str(st_path), json=str(js_path), name=name, |
| yes=getattr(args, "yes", False), |
| non_interactive=getattr(args, "non_interactive", False))) |
| finally: |
| shutil.rmtree(stage, ignore_errors=True) |
|
|
