mirror of
https://github.com/TaterTotterson/microWakeWord-Trainer-Nvidia-Docker.git
synced 2026-08-12 16:05:34 -06:00
Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
19ee63a65b |
@@ -1,3 +1,3 @@
|
|||||||
- Fixed first-run Parakeet ONNX setup failing when its empty model directory was mistaken for a complete offline model.
|
- Fixed the v19 container startup failure caused by malformed indentation in the Parakeet ONNX loader.
|
||||||
- Parakeet now downloads or resumes the required INT8 snapshot before loading through ONNX ASR.
|
- Preserved automatic download, resume, and offline reuse of the required Parakeet INT8 model snapshot.
|
||||||
- Complete local snapshots are reused without Hub access, preserving offline startup after the initial download.
|
- Revalidated both CUDA and CPU Parakeet provider paths with the complete trainer test suite.
|
||||||
|
|||||||
@@ -957,11 +957,11 @@ def _load_parakeet_onnx_model():
|
|||||||
cached = PARAKEET_ONNX_MODEL_CACHE.get(cache_key)
|
cached = PARAKEET_ONNX_MODEL_CACHE.get(cache_key)
|
||||||
if cached is not None:
|
if cached is not None:
|
||||||
return cached
|
return cached
|
||||||
suffix = (
|
suffix = (
|
||||||
f".{DEFAULT_PARAKEET_ONNX_QUANTIZATION}"
|
f".{DEFAULT_PARAKEET_ONNX_QUANTIZATION}"
|
||||||
if DEFAULT_PARAKEET_ONNX_QUANTIZATION
|
if DEFAULT_PARAKEET_ONNX_QUANTIZATION
|
||||||
else ""
|
else ""
|
||||||
)
|
)
|
||||||
model_patterns = [
|
model_patterns = [
|
||||||
"config.json",
|
"config.json",
|
||||||
"vocab.txt",
|
"vocab.txt",
|
||||||
@@ -988,10 +988,10 @@ def _load_parakeet_onnx_model():
|
|||||||
os.environ["HUGGINGFACE_HUB_CACHE"] = str(AUTO_TRAIN_MODEL_DIR / "hub")
|
os.environ["HUGGINGFACE_HUB_CACHE"] = str(AUTO_TRAIN_MODEL_DIR / "hub")
|
||||||
try:
|
try:
|
||||||
snapshot_root = AUTO_TRAIN_MODEL_DIR
|
snapshot_root = AUTO_TRAIN_MODEL_DIR
|
||||||
if not all(
|
if not all(
|
||||||
(AUTO_TRAIN_MODEL_DIR / filename).is_file()
|
(AUTO_TRAIN_MODEL_DIR / filename).is_file()
|
||||||
for filename in required_model_files
|
for filename in required_model_files
|
||||||
):
|
):
|
||||||
from huggingface_hub import snapshot_download
|
from huggingface_hub import snapshot_download
|
||||||
|
|
||||||
snapshot_root = Path(
|
snapshot_root = Path(
|
||||||
|
|||||||
Reference in New Issue
Block a user