| """Fetch Breeze inference code + suite handler without apt/curl. |
| |
| The pytorch runtime image has no curl, and apt-get ca-certificates dies |
| with exit 22 on HF Inference Endpoints. urllib is already on the image. |
| """ |
| from __future__ import annotations |
|
|
| import io |
| import os |
| import shutil |
| import tarfile |
| import urllib.request |
|
|
| DEST = "/opt/breeze-tts" |
| CODE_TGZ = "https://codeload.github.com/breezeblue-ai/breeze-tts/tar.gz/refs/heads/main" |
| HANDLER = "https://huggingface.co/copperscout1/breeze-tts-2-handler/resolve/main" |
| FILES = ("serve.py", "generate.py", "handler.py") |
|
|
|
|
| def fetch(url: str) -> bytes: |
| req = urllib.request.Request(url, headers={"User-Agent": "vividlex-breeze-tts"}) |
| with urllib.request.urlopen(req, timeout=120) as res: |
| return res.read() |
|
|
|
|
| def main() -> None: |
| os.makedirs(DEST, exist_ok=True) |
| raw = fetch(CODE_TGZ) |
| with tarfile.open(fileobj=io.BytesIO(raw), mode="r:gz") as tar: |
| tar.extractall("/tmp") |
| src = "/tmp/breeze-tts-main" |
| if not os.path.isdir(src): |
| names = [n for n in os.listdir("/tmp") if n.startswith("breeze-tts")] |
| if not names: |
| raise RuntimeError("breeze-tts tarball did not unpack") |
| src = os.path.join("/tmp", names[0]) |
| for name in os.listdir(src): |
| dest = os.path.join(DEST, name) |
| if os.path.exists(dest): |
| if os.path.isdir(dest): |
| shutil.rmtree(dest) |
| else: |
| os.remove(dest) |
| shutil.move(os.path.join(src, name), dest) |
| for name in FILES: |
| path = os.path.join(DEST, name) |
| with open(path, "wb") as fh: |
| fh.write(fetch(f"{HANDLER}/{name}")) |
| print(f"wrote {path}", flush=True) |
| print("bootstrap ok", flush=True) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
