verseflow_studio / scripts /benchmark.py
julius119's picture
Upload folder using huggingface_hub
8ab53f1 verified
Raw
History Blame Contribute Delete
1 kB
"""
VerseFlow Studio v5.0 - 硬件基准测试工具
Author: XiaoZhe (Commercial Contact: janejulius119@gmail.com / WeChat: julius119)
"""
import time
import logging
from verseflow.core.router import ModelRouter, RoutingRequest
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger("VerseFlow.Benchmark")
def run_benchmark():
logger.info("==========================================================")
logger.info(" VerseFlow Studio v5.0 硬件性能基准测试 (Benchmark Tool)")
logger.info(" Author: XiaoZhe (Commercial Contact: janejulius119@gmail.com)")
logger.info("==========================================================")
start = time.time()
router = ModelRouter()
req = RoutingRequest("product_demo", 15.0, (1280, 720), True, True, True, 30.0)
for _ in range(1000):
router.route(req)
logger.info(f"📊 模型路由器 1000 次查询总耗时: {(time.time() - start)*1000:.2f} ms")
if __name__ == "__main__":
run_benchmark()