42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
直接测试ModelManager功能
|
|
"""
|
|
import sys
|
|
import os
|
|
|
|
# 添加server目录到路径
|
|
server_path = os.path.join(os.path.dirname(__file__), 'server')
|
|
sys.path.insert(0, server_path)
|
|
|
|
from utils.model_manager import ModelManager
|
|
|
|
def test_model_manager():
|
|
print("=== 直接测试ModelManager ===")
|
|
|
|
# 创建ModelManager实例
|
|
model_manager = ModelManager()
|
|
|
|
# 获取模型列表
|
|
models = model_manager.list_models()
|
|
|
|
print(f"找到模型数量: {len(models)}")
|
|
|
|
for i, model in enumerate(models):
|
|
print(f"\n--- 模型 {i+1} (直接从ModelManager) ---")
|
|
|
|
# 显示所有字段
|
|
for key, value in model.items():
|
|
print(f"{key}: '{value}' ({type(value).__name__})")
|
|
|
|
# 特别检查filename
|
|
filename = model.get('filename', 'MISSING')
|
|
if filename != 'MISSING':
|
|
print(f"✓ filename字段存在: {filename}")
|
|
expected_model_id = filename.replace('.pth', '')
|
|
print(f"✓ 期望model_id: {expected_model_id}")
|
|
else:
|
|
print("✗ filename字段缺失!")
|
|
|
|
if __name__ == "__main__":
|
|
test_model_manager() |