2026-01-29 17:34:59 +08:00
|
|
|
<!DOCTYPE html>
|
|
|
|
|
<html lang="zh-CN">
|
|
|
|
|
<head>
|
|
|
|
|
<meta charset="UTF-8">
|
|
|
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
2026-01-29 23:10:21 +08:00
|
|
|
<title>合并权重 - YG_FT</title>
|
|
|
|
|
<link href="../../assets/libs/font-awesome-4.7.0/css/font-awesome.min.css" rel="stylesheet">
|
|
|
|
|
<script src="../../assets/libs/tailwindcss.js"></script>
|
2026-01-29 17:34:59 +08:00
|
|
|
<script>
|
2026-01-29 23:10:21 +08:00
|
|
|
tailwind.config = {
|
|
|
|
|
theme: {
|
|
|
|
|
extend: {
|
|
|
|
|
colors: {
|
|
|
|
|
primary: '#1890ff'
|
|
|
|
|
}
|
2026-01-29 17:34:59 +08:00
|
|
|
}
|
2026-01-29 23:10:21 +08:00
|
|
|
}
|
2026-01-29 17:34:59 +08:00
|
|
|
}
|
|
|
|
|
</script>
|
2026-02-02 09:22:52 +08:00
|
|
|
<script>
|
|
|
|
|
// 设置当前页面，供侧边栏高亮使用
|
|
|
|
|
window.sidebarCurrentPage = 'model-manage';
|
|
|
|
|
</script>
|
|
|
|
|
<!-- 侧边栏加载器 -->
|
|
|
|
|
<script src="../../js/components/sidebar-loader.js"></script>
|
2026-01-29 17:34:59 +08:00
|
|
|
<style>
|
2026-01-29 23:10:21 +08:00
|
|
|
body {
|
|
|
|
|
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
|
2026-01-29 17:34:59 +08:00
|
|
|
}
|
|
|
|
|
</style>
|
|
|
|
|
</head>
|
2026-01-29 23:10:21 +08:00
|
|
|
<body class="bg-gray-100 h-screen flex overflow-hidden">
|
2026-02-02 09:22:52 +08:00
|
|
|
<!-- 侧边栏容器 -->
|
|
|
|
|
<div id="sidebar-container"></div>
|
2026-01-29 17:34:59 +08:00
|
|
|
|
2026-02-02 09:22:52 +08:00
|
|
|
<!-- 主内容区 -->
|
2026-01-29 23:10:21 +08:00
|
|
|
<div class="sidebar-section-title mt-6">训练管理</div>
|
2026-01-29 17:34:59 +08:00
|
|
|
<div class="nav-item-wrapper">
|
2026-01-29 23:10:21 +08:00
|
|
|
<a href="main.html?page=fine-tune" data-page="fine-tune" class="nav-link flex items-center px-4 py-2.5 hover:bg-[#001529]/20 transition-colors">
|
|
|
|
|
<i class="fa fa-magic w-5 text-center"></i>
|
|
|
|
|
<span class="ml-2">训练任务</span>
|
2026-01-29 17:34:59 +08:00
|
|
|
</a>
|
|
|
|
|
</div>
|
|
|
|
|
<div class="nav-item-wrapper">
|
2026-01-29 23:10:21 +08:00
|
|
|
<a href="main.html?page=data-manage" data-page="data-manage" class="nav-link flex items-center px-4 py-2.5 hover:bg-[#001529]/20 transition-colors">
|
|
|
|
|
<i class="fa fa-database w-5 text-center"></i>
|
2026-01-29 17:34:59 +08:00
|
|
|
<span class="ml-2">数据集管理</span>
|
|
|
|
|
</a>
|
|
|
|
|
</div>
|
|
|
|
|
<div class="nav-item-wrapper">
|
|
|
|
|
<a href="#" data-page="data-generate" class="nav-link flex items-center px-4 py-2.5 hover:bg-[#001529]/20 transition-colors">
|
|
|
|
|
<i class="fa fa-database w-5 text-center"></i>
|
|
|
|
|
<span class="ml-2">其他工具</span>
|
|
|
|
|
</a>
|
|
|
|
|
</div>
|
|
|
|
|
|
|
|
|
|
<!-- 第三分区：系统设置 -->
|
|
|
|
|
<div class="sidebar-section-title mt-6">系统设置</div>
|
|
|
|
|
<div class="nav-item-wrapper">
|
|
|
|
|
<a href="#" data-page="config" class="nav-link flex items-center px-4 py-2.5 hover:bg-[#001529]/20 transition-colors">
|
|
|
|
|
<i class="fa fa-bar-chart w-5 text-center"></i>
|
|
|
|
|
<span class="ml-2">平台性能</span>
|
|
|
|
|
</a>
|
|
|
|
|
</div>
|
|
|
|
|
<div class="nav-item-wrapper">
|
|
|
|
|
<a href="main.html?page=logs" data-page="logs" class="nav-link flex items-center px-4 py-2.5 hover:bg-[#001529]/20 transition-colors">
|
|
|
|
|
<i class="fa fa-file-text w-5 text-center"></i>
|
|
|
|
|
<span class="ml-2">查看日志</span>
|
|
|
|
|
</a>
|
|
|
|
|
</div>
|
|
|
|
|
</nav>
|
|
|
|
|
|
|
|
|
|
<div class="p-4 border-t border-[#001529]/30 text-xs mt-auto">
|
|
|
|
|
<div class="mb-2 text-[#bfcbd9]/80">默认业务空间</div>
|
|
|
|
|
<div class="flex items-center justify-between">
|
|
|
|
|
<span class="text-[#bfcbd9]">版本 v1.0.0</span>
|
|
|
|
|
</div>
|
|
|
|
|
</div>
|
|
|
|
|
</aside>
|
|
|
|
|
|
|
|
|
|
<!-- 主内容区 -->
|
|
|
|
|
<div class="flex-1 flex flex-col overflow-hidden">
|
|
|
|
|
<!-- 顶部导航 -->
|
|
|
|
|
<header class="bg-white border-b border-gray-200 shadow-sm">
|
|
|
|
|
<div class="flex items-center justify-between px-6 h-14">
|
|
|
|
|
<div class="flex items-center space-x-4">
|
|
|
|
|
<a href="main.html?page=my-models" class="text-gray-500 hover:text-gray-700 flex items-center">
|
|
|
|
|
<i class="fa fa-arrow-left"></i>
|
|
|
|
|
<span class="ml-1">返回我的模型</span>
|
|
|
|
|
</a>
|
|
|
|
|
</div>
|
|
|
|
|
<div class="flex items-center space-x-4">
|
|
|
|
|
<div class="flex items-center text-sm text-gray-600">
|
|
|
|
|
<i class="fa fa-cube text-primary mr-2"></i>
|
|
|
|
|
<span id="modelTitle">加载中...</span>
|
|
|
|
|
</div>
|
|
|
|
|
</div>
|
|
|
|
|
</div>
|
|
|
|
|
</header>
|
|
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
<!-- 主内容 -->
|
2026-01-29 17:34:59 +08:00
|
|
|
<main class="flex-1 flex flex-col overflow-hidden bg-gray-50">
|
|
|
|
|
<!-- 模型信息栏 -->
|
|
|
|
|
<div class="bg-white border-b border-gray-200 px-6 py-3 flex items-center justify-between">
|
|
|
|
|
<div class="flex items-center space-x-6">
|
|
|
|
|
<div class="flex items-center">
|
|
|
|
|
<span class="text-xs text-gray-400 mr-2">训练方法:</span>
|
|
|
|
|
<span id="trainMethod" class="px-2 py-1 bg-green-100 text-green-700 text-xs rounded">-</span>
|
|
|
|
|
</div>
|
|
|
|
|
<div class="flex items-center">
|
|
|
|
|
<span class="text-xs text-gray-400 mr-2">基座模型:</span>
|
|
|
|
|
<span id="baseModel" class="text-sm text-gray-700">-</span>
|
|
|
|
|
</div>
|
|
|
|
|
</div>
|
|
|
|
|
</div>
|
|
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
<!-- 合并权重区域 -->
|
|
|
|
|
<div id="mergeContainer" class="flex-1 overflow-y-auto p-6">
|
|
|
|
|
<!-- 合并状态 -->
|
|
|
|
|
<div id="mergeState" class="flex flex-col items-center justify-center h-full text-center py-12">
|
|
|
|
|
<div class="w-16 h-16 rounded-full bg-blue-100 flex items-center justify-center mb-4">
|
|
|
|
|
<i class="fa fa-compress text-2xl text-blue-500"></i>
|
2026-01-29 17:34:59 +08:00
|
|
|
</div>
|
2026-01-29 23:10:21 +08:00
|
|
|
<h3 class="text-lg font-medium text-gray-600 mb-2">合并模型权重</h3>
|
|
|
|
|
<p class="text-sm text-gray-400 max-w-sm mb-4" id="mergeText">将LoRA适配器权重合并到基座模型</p>
|
|
|
|
|
<button onclick="mergeWeights()" id="mergeBtn"
|
|
|
|
|
class="px-6 py-2.5 bg-primary text-white rounded-lg hover:bg-primary/90 transition-colors flex items-center">
|
|
|
|
|
<i class="fa fa-compress mr-2"></i>
|
|
|
|
|
<span>开始合并</span>
|
|
|
|
|
</button>
|
2026-01-29 17:34:59 +08:00
|
|
|
</div>
|
2026-01-29 23:10:21 +08:00
|
|
|
<!-- 合并结果 -->
|
|
|
|
|
<div id="mergeResult" class="hidden flex flex-col items-center justify-center h-full text-center py-12">
|
|
|
|
|
<div class="w-16 h-16 rounded-full bg-green-100 flex items-center justify-center mb-4">
|
|
|
|
|
<i class="fa fa-check-circle text-2xl text-green-500"></i>
|
2026-01-29 17:34:59 +08:00
|
|
|
</div>
|
2026-01-29 23:10:21 +08:00
|
|
|
<h3 class="text-lg font-medium text-gray-600 mb-2">合并完成</h3>
|
|
|
|
|
<p class="text-sm text-gray-400 max-w-sm mb-4" id="resultText"></p>
|
|
|
|
|
<button onclick="location.reload()"
|
|
|
|
|
class="px-6 py-2.5 bg-primary text-white rounded-lg hover:bg-primary/90 transition-colors">
|
|
|
|
|
<i class="fa fa-refresh mr-2"></i>
|
|
|
|
|
<span>返回</span>
|
|
|
|
|
</button>
|
2026-01-29 17:34:59 +08:00
|
|
|
</div>
|
|
|
|
|
</div>
|
|
|
|
|
</main>
|
|
|
|
|
</div>
|
|
|
|
|
|
|
|
|
|
<script>
|
|
|
|
|
// API 基础地址
|
|
|
|
|
const getApiBase = () => {
|
|
|
|
|
const protocol = window.location.protocol;
|
|
|
|
|
const hostname = window.location.hostname;
|
|
|
|
|
return `${protocol}//${hostname}:7861/api`;
|
|
|
|
|
};
|
|
|
|
|
const API_BASE = typeof window.API_BASE !== 'undefined' ? window.API_BASE : getApiBase();
|
|
|
|
|
|
|
|
|
|
// 全局状态
|
|
|
|
|
let currentModel = {
|
|
|
|
|
name: '',
|
|
|
|
|
trainMethod: '',
|
|
|
|
|
baseModel: ''
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// 页面初始化
|
2026-01-29 23:10:21 +08:00
|
|
|
document.addEventListener('DOMContentLoaded', async function() {
|
2026-01-29 17:34:59 +08:00
|
|
|
// 解析URL参数
|
|
|
|
|
const urlParams = new URLSearchParams(window.location.search);
|
|
|
|
|
const modelName = urlParams.get('model');
|
|
|
|
|
const trainMethod = urlParams.get('method');
|
|
|
|
|
|
|
|
|
|
if (modelName) {
|
|
|
|
|
currentModel.name = decodeURIComponent(modelName);
|
|
|
|
|
document.getElementById('modelTitle').textContent = currentModel.name;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (trainMethod) {
|
|
|
|
|
currentModel.trainMethod = decodeURIComponent(trainMethod);
|
|
|
|
|
const methodDisplay = {
|
|
|
|
|
'lora': 'LoRA',
|
|
|
|
|
'full': '全参微调',
|
|
|
|
|
'sft': 'SFT',
|
|
|
|
|
'dpo': 'DPO',
|
|
|
|
|
'cpt': 'CPT'
|
|
|
|
|
};
|
|
|
|
|
document.getElementById('trainMethod').textContent = methodDisplay[currentModel.trainMethod] || currentModel.trainMethod;
|
|
|
|
|
}
|
|
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
// 加载模型信息
|
|
|
|
|
if (currentModel.name) {
|
|
|
|
|
await loadModelInfo();
|
|
|
|
|
}
|
|
|
|
|
|
2026-01-29 17:34:59 +08:00
|
|
|
// 绑定侧边栏导航点击事件
|
|
|
|
|
document.querySelectorAll('.nav-link').forEach(link => {
|
|
|
|
|
link.addEventListener('click', function(e) {
|
|
|
|
|
e.preventDefault();
|
|
|
|
|
const page = this.dataset.page;
|
|
|
|
|
if (page === 'logs') {
|
|
|
|
|
window.location.href = `main.html?page=${page}`;
|
|
|
|
|
} else if (page) {
|
|
|
|
|
window.location.href = `main.html?page=${page}`;
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
if (link.dataset.page === 'my-models') {
|
|
|
|
|
link.classList.add('sidebar-item-active');
|
|
|
|
|
link.classList.remove('hover:bg-[#001529]/20');
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
updateSidebarSlider();
|
2026-01-29 23:10:21 +08:00
|
|
|
});
|
2026-01-29 17:34:59 +08:00
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
// 侧边栏高亮
|
|
|
|
|
function updateSidebarSlider() {
|
|
|
|
|
const activeItem = document.querySelector('.sidebar-item-active');
|
|
|
|
|
const slider = document.getElementById('sidebarSlider');
|
|
|
|
|
const wrapper = document.querySelector('.nav-item-wrapper');
|
|
|
|
|
if (activeItem && slider && wrapper) {
|
|
|
|
|
slider.style.display = 'block';
|
|
|
|
|
slider.style.width = wrapper.offsetWidth + 'px';
|
|
|
|
|
slider.style.top = wrapper.offsetTop + 'px';
|
|
|
|
|
slider.style.height = wrapper.offsetHeight + 'px';
|
|
|
|
|
}
|
|
|
|
|
}
|
2026-01-29 17:34:59 +08:00
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
// 加载模型信息
|
|
|
|
|
async function loadModelInfo() {
|
|
|
|
|
try {
|
|
|
|
|
const response = await fetch(`${API_BASE}/model-manage/trained-models`);
|
|
|
|
|
const result = await response.json();
|
2026-01-29 17:34:59 +08:00
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
if (result.code !== 0) {
|
|
|
|
|
console.error('[DEBUG] 获取模型列表失败');
|
|
|
|
|
return;
|
|
|
|
|
}
|
2026-01-29 17:34:59 +08:00
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
const models = result.data?.models || [];
|
|
|
|
|
const modelInfo = models.find(m => m.name === currentModel.name);
|
|
|
|
|
console.log('[DEBUG] 模型查找:', currentModel.name, modelInfo);
|
2026-01-29 17:34:59 +08:00
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
if (modelInfo) {
|
|
|
|
|
currentModel.baseModel = modelInfo.base_model_path || '';
|
|
|
|
|
document.getElementById('baseModel').textContent = currentModel.baseModel || '未知';
|
|
|
|
|
} else {
|
|
|
|
|
document.getElementById('baseModel').textContent = '未知';
|
|
|
|
|
}
|
|
|
|
|
} catch (error) {
|
|
|
|
|
console.error('[DEBUG] 加载模型信息失败:', error);
|
|
|
|
|
document.getElementById('baseModel').textContent = '未知';
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 合并权重
|
|
|
|
|
async function mergeWeights() {
|
2026-01-29 17:34:59 +08:00
|
|
|
if (!currentModel.name) {
|
2026-01-29 23:10:21 +08:00
|
|
|
alert('模型参数无效');
|
2026-01-29 17:34:59 +08:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
const mergeBtn = document.getElementById('mergeBtn');
|
|
|
|
|
const mergeText = document.getElementById('mergeText');
|
|
|
|
|
const originalBtnText = mergeBtn.innerHTML;
|
2026-01-29 17:34:59 +08:00
|
|
|
|
2026-01-29 23:10:21 +08:00
|
|
|
mergeBtn.disabled = true;
|
|
|
|
|
mergeBtn.innerHTML = '<i class="fa fa-spinner fa-spin mr-2"></i>合并中...';
|
|
|
|
|
mergeText.textContent = '正在合并模型权重，请稍候...';
|
2026-01-29 17:34:59 +08:00
|
|
|
|
|
|
|
|
try {
|
2026-01-29 23:10:21 +08:00
|
|
|
const response = await fetch(`${API_BASE}/model-manage/merge`, {
|
2026-01-29 17:34:59 +08:00
|
|
|
method: 'POST',
|
|
|
|
|
headers: { 'Content-Type': 'application/json' },
|
|
|
|
|
body: JSON.stringify({
|
|
|
|
|
model_name: currentModel.name,
|
|
|
|
|
train_method: currentModel.trainMethod || 'lora',
|
2026-01-29 23:10:21 +08:00
|
|
|
base_model_path: currentModel.baseModel
|
2026-01-29 17:34:59 +08:00
|
|
|
})
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
const result = await response.json();
|
2026-01-29 23:10:21 +08:00
|
|
|
console.log('[DEBUG] 合并响应:', result);
|
2026-01-29 17:34:59 +08:00
|
|
|
|
|
|
|
|
if (result.code === 0) {
|
2026-01-29 23:10:21 +08:00
|
|
|
mergeText.textContent = '合并成功！';
|
|
|
|
|
document.getElementById('mergeState').classList.add('hidden');
|
|
|
|
|
document.getElementById('mergeResult').classList.remove('hidden');
|
|
|
|
|
document.getElementById('resultText').textContent = result.message || '模型权重已成功合并';
|
2026-01-29 17:34:59 +08:00
|
|
|
} else {
|
2026-01-29 23:10:21 +08:00
|
|
|
mergeText.textContent = result.message || '合并失败';
|
|
|
|
|
mergeBtn.disabled = false;
|
|
|
|
|
mergeBtn.innerHTML = originalBtnText;
|
2026-01-29 17:34:59 +08:00
|
|
|
}
|
|
|
|
|
} catch (error) {
|
2026-01-29 23:10:21 +08:00
|
|
|
console.error('[DEBUG] 合并失败:', error);
|
|
|
|
|
mergeText.textContent = '合并失败: ' + error.message;
|
|
|
|
|
mergeBtn.disabled = false;
|
|
|
|
|
mergeBtn.innerHTML = originalBtnText;
|
2026-01-29 17:34:59 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 暴露到全局
|
2026-01-29 23:10:21 +08:00
|
|
|
window.mergeWeights = mergeWeights;
|
2026-01-29 17:34:59 +08:00
|
|
|
</script>
|
|
|
|
|
</body>
|
|
|
|
|
</html>
|
