Spaces:
Runtime error
Runtime error
File size: 881 Bytes
a4a3878 d344109 a4a3878 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | from __future__ import annotations
import argparse
from pathlib import Path
from faster_whisper import WhisperModel
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Download faster-whisper model")
parser.add_argument("--model", default="small", help="Model size/id (tiny, base, small)")
parser.add_argument("--output", default="./models", help="Output directory")
parser.add_argument("--device", default="cpu")
parser.add_argument("--compute-type", default="int8")
return parser.parse_args()
def main() -> None:
args = parse_args()
output = Path(args.output)
output.mkdir(parents=True, exist_ok=True)
WhisperModel(
args.model,
device=args.device,
compute_type=args.compute_type,
download_root=str(output),
)
print(f"Model {args.model} cached under {output}")
if __name__ == "__main__":
main()
|