👍️[Update] Model : status 348のlog を printLog関数に集約

This commit is contained in:
misyaguziya
2024-09-01 17:33:22 +09:00
parent 8f2c81a7d7
commit 4c2e4a6e04
7 changed files with 159 additions and 143 deletions

View File

@@ -1,9 +1,13 @@
from os import path as os_path, makedirs as os_makedirs
import sys
sys.path.append(os_path.join(os_path.dirname(__file__), "..", ".."))
from utils import printLog
from requests import get as requests_get
from typing import Callable
import huggingface_hub
from faster_whisper import WhisperModel
import logging
logger = logging.getLogger('faster_whisper')
logger.setLevel(logging.CRITICAL)
@@ -33,15 +37,14 @@ def downloadFile(url, path, func=None):
file_size = int(res.headers.get('content-length', 0))
total_chunk = 0
with open(os_path.join(path), 'wb') as file:
for chunk in res.iter_content(chunk_size=1024*5):
for chunk in res.iter_content(chunk_size=1024*2000):
file.write(chunk)
if isinstance(func, Callable):
total_chunk += len(chunk)
func(total_chunk/file_size)
except Exception as e:
import json
print(json.dumps({"status":348, "log": f"error:downloadFile() {e}"}), flush=True)
printLog("warning:downloadFile()", e)
def checkWhisperWeight(root, weight_type):
path = os_path.join(root, "weights", "whisper", weight_type)