nextrec 0.5.4__tar.gz → 0.5.5__tar.gz

{nextrec-0.5.4 → nextrec-0.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.5.4
+Version: 0.5.5
 Summary: A comprehensive recommendation library with match, ranking, and multi-task learning models
 Project-URL: Homepage, https://github.com/zerolovesea/NextRec
 Project-URL: Repository, https://github.com/zerolovesea/NextRec
@@ -73,7 +73,7 @@ Description-Content-Type: text/markdown
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.5.4-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.5.5-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -259,11 +259,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.5.4，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.5.5，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.5.4，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.5.5，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.5.4 → nextrec-0.5.5}/README.md

@@ -8,7 +8,7 @@
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.5.4-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.5.5-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -194,11 +194,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.5.4，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.5.5，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.5.4，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.5.5，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.5.4 → nextrec-0.5.5}/README_en.md

@@ -8,7 +8,7 @@
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.5.4-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.5.5-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 English | [中文文档](README.md)
@@ -197,11 +197,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 Prediction outputs are saved under `{checkpoint_path}/predictions/{name}.{save_data_format}`.
 
-> As of version 0.5.4, NextRec CLI supports single-machine training; distributed training features are currently under development.
+> As of version 0.5.5, NextRec CLI supports single-machine training; distributed training features are currently under development.
 
 ## Platform Compatibility
 
-The current version is 0.5.4. All models and test code have been validated on the following platforms. If you encounter compatibility issues, please report them in the issue tracker with your system version:
+The current version is 0.5.5. All models and test code have been validated on the following platforms. If you encounter compatibility issues, please report them in the issue tracker with your system version:
 
 | Platform | Configuration | 
 |----------|---------------|

{nextrec-0.5.4 → nextrec-0.5.5}/docs/rtd/conf.py

@@ -11,7 +11,7 @@ sys.path.insert(0, str(PROJECT_ROOT / "nextrec"))
 project = "NextRec"
 copyright = "2026, Yang Zhou"
 author = "Yang Zhou"
-release = "0.5.4"
+release = "0.5.5"
 
 extensions = [
     "myst_parser",

nextrec-0.5.5/nextrec/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.5.5"

{nextrec-0.5.4 → nextrec-0.5.5}/nextrec/cli.py

@@ -137,7 +137,7 @@ def train_model(train_config_path: str) -> None:
     model_cfg = read_yaml(model_cfg_path)
 
     # Extract id_column from data config for GAUC metrics
-    id_column = data_cfg.get("id_column") or data_cfg.get("user_id_column")
+    id_column = data_cfg.get("id_column")
     id_columns = [id_column] if id_column else []
 
     log_cli_section("Data")
@@ -378,6 +378,7 @@ def train_model(train_config_path: str) -> None:
         sparse_features,
         sequence_features,
         target,
+        id_columns,
         device,
     )
 
@@ -585,14 +586,16 @@ def predict_model(predict_config_path: str) -> None:
         sparse_features=sparse_features,
         sequence_features=sequence_features,
         target=target_cols,
+        id_columns=id_columns,
         device=predict_cfg.get("device", "cpu"),
     )
-    model.id_columns = id_columns
+
     model.load_model(
         model_file, map_location=predict_cfg.get("device", "cpu"), verbose=True
     )
 
-    effective_id_columns = model.id_columns
+    effective_id_columns = id_columns or model.id_columns
+
     log_cli_section("Features")
     log_kv_lines(
         [

{nextrec-0.5.4 → nextrec-0.5.5}/nextrec/utils/config.py

@@ -388,6 +388,7 @@ def build_model_instance(
     sparse_features: List["SparseFeature"],
     sequence_features: List["SequenceFeature"],
     target: List[str],
+    id_columns: list[str] | str | None,
     device: str,
 ) -> Any:
     """
@@ -400,6 +401,7 @@ def build_model_instance(
         sparse_features: List of sparse feature objects
         sequence_features: List of sequence feature objects
         target: List of target column names
+        id_columns: Identifier column name(s) for GAUC or ID passthrough
         device: Device string (e.g., 'cpu', 'cuda:0')
     """
     dense_map = {f.name: f for f in dense_features}
@@ -436,7 +438,7 @@ def build_model_instance(
         param.kind == inspect.Parameter.VAR_KEYWORD for param in sig_params.values()
     )
 
-    init_kwargs: Dict[str, Any] = dict(params_cfg)
+    init_kwargs = dict(params_cfg)
 
     # Explicit bindings (model_config.feature_bindings) take priority
     for param_name, binding in feature_bindings_cfg.items():
@@ -517,6 +519,8 @@ def build_model_instance(
 
     if accepts("target"):
         init_kwargs.setdefault("target", target)
+    if accepts("id_columns") or accepts_var_kwargs:
+        init_kwargs.setdefault("id_columns", id_columns)
     if accepts("device"):
         init_kwargs.setdefault("device", device)
 

{nextrec-0.5.4 → nextrec-0.5.5}/nextrec_cli_preset/predict_config.yaml

@@ -6,6 +6,8 @@ predict:
   data_path: ../dataset/ctcvr_task.csv
   source_data_format: csv                  # Input data format: csv, parquet, etc.
                                            # Use 'auto' for automatic detection
+  id_column: user_id                       # Column name for user/sample IDs
+                                           # Will be included in prediction output
   name: pred                               # Output filename (without extension) or absolute path
                                            # Filename -> {checkpoint_path}/predictions/{name}.{save_data_format}
                                            # Absolute path -> saved directly (extension auto-added if missing)

{nextrec-0.5.4 → nextrec-0.5.5}/nextrec_cli_preset/predict_config_template.yaml

@@ -30,6 +30,10 @@ predict:
   # ----- Data Format Settings -----
   source_data_format: parquet              # Input data format: csv, parquet, etc.
                                            # Use 'auto' for automatic detection
+
+  # ----- Identification Settings -----
+  id_column: user_id                       # Column name for user/sample IDs
+                                           # Will be included in prediction output
   
   # ----- Output Settings -----
   name: pred                               # Output filename (without extension) or absolute path

{nextrec-0.5.4 → nextrec-0.5.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nextrec"
-version = "0.5.4"
+version = "0.5.5"
 description = "A comprehensive recommendation library with match, ranking, and multi-task learning models"
 readme = "README.md"
 requires-python = ">=3.10"

{nextrec-0.5.4 → nextrec-0.5.5}/ui/nextrec-ui/src/pages/FeaturePage.vue

@@ -8,7 +8,7 @@
     </div>
 
     <div class="card">
-      <h2>{{ t('Dense 特征', 'Dense Features') }}</h2>
+      <h2>{{ t('稠密特征', 'Dense Features') }}</h2>
       <div class="inline-list">
         <div v-for="(item, index) in dense" :key="item.id" class="inline-item">
           <div class="grid-inline">
@@ -17,11 +17,13 @@
               <input v-model="item.name" placeholder="user_active_days_7" />
             </div>
             <div class="field">
-              <label>{{ t('归一化方式', 'scaler') }}</label>
+              <label>{{ t('缩放方式', 'Scaler') }}</label>
               <select v-model="item.processor.scaler">
                 <option value="standard">standard</option>
                 <option value="minmax">minmax</option>
                 <option value="robust">robust</option>
+                <option value="maxabs">maxabs</option>
+                <option value="log">log</option>
                 <option value="none">none</option>
               </select>
             </div>
@@ -42,7 +44,7 @@
             </div>
           </div>
           <div class="actions">
-            <button class="ghost" @click="remove(dense, index)">{{ t('移除', 'Remove') }}</button>
+            <button class="icon-button icon-button-small icon-button-danger" @click="remove(dense, index)" aria-label="移除">×</button>
           </div>
         </div>
       </div>
@@ -52,7 +54,7 @@
     </div>
 
     <div class="card">
-      <h2>{{ t('Sparse 特征', 'Sparse Features') }}</h2>
+      <h2>{{ t('稀疏特征', 'Sparse Features') }}</h2>
       <div class="inline-list">
         <div v-for="(item, index) in sparse" :key="item.id" class="inline-item">
           <div class="grid-inline">
@@ -68,10 +70,10 @@
               <label>{{ t('编码方式', 'encode_method') }}</label>
               <select v-model="item.processor.encode_method">
                 <option value="hash">hash</option>
-                <option value="vocab">vocab</option>
+                <option value="label">label</option>
               </select>
             </div>
-            <div class="field">
+            <div class="field" v-if="item.processor.encode_method === 'hash'">
               <label>{{ t('哈希桶大小', 'hash_size') }}</label>
               <input v-model.number="item.processor.hash_size" type="number" />
             </div>
@@ -89,7 +91,7 @@
             </div>
           </div>
           <div class="actions">
-            <button class="ghost" @click="remove(sparse, index)">{{ t('移除', 'Remove') }}</button>
+            <button class="icon-button icon-button-small icon-button-danger" @click="remove(sparse, index)" aria-label="移除">×</button>
           </div>
         </div>
       </div>
@@ -99,7 +101,7 @@
     </div>
 
     <div class="card">
-      <h2>{{ t('Sequence 特征', 'Sequence Features') }}</h2>
+      <h2>{{ t('序列特征', 'Sequence Features') }}</h2>
       <div class="inline-list">
         <div v-for="(item, index) in sequence" :key="item.id" class="inline-item">
           <div class="grid-inline">
@@ -115,10 +117,10 @@
               <label>{{ t('编码方式', 'encode_method') }}</label>
               <select v-model="item.processor.encode_method">
                 <option value="hash">hash</option>
-                <option value="vocab">vocab</option>
+                <option value="label">label</option>
               </select>
             </div>
-            <div class="field">
+            <div class="field" v-if="item.processor.encode_method === 'hash'">
               <label>{{ t('哈希桶大小', 'hash_size') }}</label>
               <input v-model.number="item.processor.hash_size" type="number" />
             </div>
@@ -127,7 +129,7 @@
               <input v-model.number="item.processor.min_freq" type="number" />
             </div>
             <div class="field">
-              <label>{{ t('最大长度', 'max_len') }}</label>
+              <label>{{ t('截断最大长度', 'max_len') }}</label>
               <input v-model.number="item.processor.max_len" type="number" />
             </div>
             <div class="field">
@@ -149,10 +151,6 @@
               <label>{{ t('词表大小', 'vocab_size') }}</label>
               <input v-model.number="item.embedding.vocab_size" type="number" />
             </div>
-            <div class="field">
-              <label>{{ t('最大长度', 'max_len') }}</label>
-              <input v-model.number="item.embedding.max_len" type="number" />
-            </div>
             <div class="field">
               <label>{{ t('池化方式', 'combiner') }}</label>
               <select v-model="item.embedding.combiner">
@@ -173,7 +171,7 @@
             </div>
           </div>
           <div class="actions">
-            <button class="ghost" @click="remove(sequence, index)">{{ t('移除', 'Remove') }}</button>
+            <button class="icon-button icon-button-small icon-button-danger" @click="remove(sequence, index)" aria-label="移除">×</button>
           </div>
         </div>
       </div>
@@ -188,7 +186,7 @@
         <textarea class="mono" readonly :value="yamlText"></textarea>
       </div>
       <div class="actions spaced-top">
-        <button class="primary" @click="download">{{ t('下载 feature_config.yaml', 'Download feature_config.yaml') }}</button>
+        <button class="primary" @click="download">{{ t('下载', 'Download') }}</button>
       </div>
     </div>
   </section>
@@ -230,6 +228,7 @@ function buildSection(items) {
       continue;
     }
     const embedding = { ...item.embedding };
+    embedding.max_len = item.processor.max_len;
     embedding.name = item.name;
     section[item.name] = {
       processor_config: { ...item.processor },

{nextrec-0.5.4 → nextrec-0.5.5}/ui/nextrec-ui/src/pages/ModelPage.vue

@@ -17,27 +17,8 @@
           </select>
         </div>
       </div>
-    </div>
-
-    <div class="card">
-      <h2>{{ t('正则配置', 'Regularization') }}</h2>
-      <div class="grid-inline">
-        <div class="field">
-          <label>embedding_l1_reg</label>
-          <input v-model.number="regEmbeddingL1" type="number" step="0.000001" />
-        </div>
-        <div class="field">
-          <label>embedding_l2_reg</label>
-          <input v-model.number="regEmbeddingL2" type="number" step="0.000001" />
-        </div>
-        <div class="field">
-          <label>dense_l1_reg</label>
-          <input v-model.number="regDenseL1" type="number" step="0.000001" />
-        </div>
-        <div class="field">
-          <label>dense_l2_reg</label>
-          <input v-model.number="regDenseL2" type="number" step="0.000001" />
-        </div>
+      <div v-if="modelNote" class="alert model-note" style="margin-top: 12px;">
+        {{ modelNote }}
       </div>
     </div>
 
@@ -52,11 +33,19 @@
             </div>
           </div>
           <div class="field" style="margin-top: 12px;">
-            <label>{{ t('值', 'Value') }}</label>
+            <label>
+              {{ t('值', 'Value') }}
+              <span
+                v-if="paramHelp[entry.key]"
+                class="help-icon"
+                :title="paramHelp[entry.key]"
+                :data-tip="paramHelp[entry.key]"
+              >?</span>
+            </label>
             <textarea v-model="entry.raw" class="mono" placeholder="relu 或 123 或 true 或 \nmlp_params:\n  hidden_dims: [256, 128]"></textarea>
           </div>
           <div class="actions">
-            <button class="icon-button" @click="removeEntryById(entry.id)" :disabled="paramEntries.length <= 1" aria-label="移除">×</button>
+            <button class="icon-button icon-button-small icon-button-danger" @click="removeEntryById(entry.id)" :disabled="paramEntries.length <= 1" aria-label="移除">×</button>
           </div>
         </div>
       </div>
@@ -65,6 +54,28 @@
       </div>
     </div>
 
+    <div class="card">
+      <h2>{{ t('正则配置', 'Regularization') }}</h2>
+      <div class="grid-inline">
+        <div class="field">
+          <label>embedding_l1_reg</label>
+          <input v-model.number="regEmbeddingL1" type="number" step="0.000001" />
+        </div>
+        <div class="field">
+          <label>embedding_l2_reg</label>
+          <input v-model.number="regEmbeddingL2" type="number" step="0.000001" />
+        </div>
+        <div class="field">
+          <label>dense_l1_reg</label>
+          <input v-model.number="regDenseL1" type="number" step="0.000001" />
+        </div>
+        <div class="field">
+          <label>dense_l2_reg</label>
+          <input v-model.number="regDenseL2" type="number" step="0.000001" />
+        </div>
+      </div>
+    </div>
+
     <div class="card">
       <h2>{{ t('YAML 预览', 'YAML Preview') }}</h2>
       <div v-if="error" class="alert">{{ error }}</div>
@@ -72,7 +83,7 @@
         <textarea class="mono" readonly :value="yamlText"></textarea>
       </div>
       <div class="actions spaced-top">
-        <button class="primary" @click="download" :disabled="!!error">{{ t('下载 model_config.yaml', 'Download model_config.yaml') }}</button>
+        <button class="primary" @click="download" :disabled="!!error">{{ t('下载', 'Download') }}</button>
       </div>
     </div>
   </section>
@@ -99,6 +110,68 @@ function toParamsText(raw) {
   return '';
 }
 
+function parsePresetComments(raw) {
+  const lines = raw.split(/\r?\n/);
+  const topComments = [];
+  let inTop = true;
+  const paramComments = {};
+  const pendingByIndent = new Map();
+  let inParams = false;
+
+  for (let i = 0; i < lines.length; i += 1) {
+    const line = lines[i];
+    const trimmed = line.trim();
+    if (inTop) {
+      if (trimmed.startsWith('#')) {
+        topComments.push(trimmed.replace(/^#+\s?/, ''));
+        continue;
+      }
+      if (trimmed === '') {
+        continue;
+      }
+      inTop = false;
+    }
+
+    if (trimmed.startsWith('params:')) {
+      inParams = true;
+      continue;
+    }
+    if (!inParams) {
+      continue;
+    }
+
+    const indent = line.match(/^\s*/)?.[0].length ?? 0;
+    if (trimmed.startsWith('#') && indent >= 2) {
+      pendingByIndent.set(indent, trimmed.replace(/^#+\s?/, ''));
+      continue;
+    }
+
+    const match = line.match(/^\s{2,}([A-Za-z0-9_]+):/);
+    if (!match) {
+      continue;
+    }
+    const key = match[1];
+    let comment = '';
+    const inlineIdx = line.indexOf('#');
+    if (inlineIdx !== -1) {
+      comment = line.slice(inlineIdx + 1).trim();
+    } else {
+      comment = pendingByIndent.get(indent) || '';
+    }
+    if (comment) {
+      paramComments[key] = comment;
+    }
+    pendingByIndent.delete(indent);
+  }
+
+  return { top: topComments.join('\n'), params: paramComments };
+}
+
+function getPresetRaw(name) {
+  const entry = Object.entries(presetModules).find(([path]) => path.endsWith(`/${name}.yaml`));
+  return entry ? entry[1] : '';
+}
+
 const presets = Object.entries(presetModules).reduce((acc, [path, raw]) => {
   const data = parseYaml(raw) || {};
   const modelName = data.model || filenameToModel(path);
@@ -116,6 +189,8 @@ const lang = computed(() => store.ui.lang);
 const t = (zh, en) => (lang.value === 'zh' ? zh : en);
 const error = ref('');
 const paramEntries = reactive([]);
+const paramHelp = reactive({});
+const modelNote = ref('');
 const hiddenParamKeys = new Set(['embedding_l1_reg', 'embedding_l2_reg', 'dense_l1_reg', 'dense_l2_reg']);
 const visibleParamEntries = computed(() => paramEntries.filter((entry) => !hiddenParamKeys.has(entry.key)));
 
@@ -159,6 +234,12 @@ function applyPreset() {
   form.paramsText = preset.paramsText;
   const params = parseYaml(form.paramsText) || {};
   setEntriesFromParams(params);
+  const comments = parsePresetComments(getPresetRaw(modelName));
+  modelNote.value = comments.top;
+  Object.keys(paramHelp).forEach((key) => delete paramHelp[key]);
+  Object.entries(comments.params).forEach(([key, note]) => {
+    paramHelp[key] = note;
+  });
 }
 
 function addEntry() {
@@ -272,5 +353,13 @@ onMounted(() => {
   }
   const params = parseYaml(form.paramsText || '') || {};
   setEntriesFromParams(params);
+  if (preset) {
+    const comments = parsePresetComments(getPresetRaw(preset.model));
+    modelNote.value = comments.top;
+    Object.keys(paramHelp).forEach((key) => delete paramHelp[key]);
+    Object.entries(comments.params).forEach(([key, note]) => {
+      paramHelp[key] = note;
+    });
+  }
 });
 </script>

{nextrec-0.5.4 → nextrec-0.5.5}/ui/nextrec-ui/src/pages/PredictPage.vue

@@ -29,11 +29,11 @@
       <h2>{{ t('Predict 参数', 'Prediction Settings') }}</h2>
       <div class="grid-2">
         <div class="field">
-          <label>{{ t('Data Path', 'Data Path') }}</label>
+          <label>{{ t('数据路径', 'Data Path') }}</label>
           <input v-model="form.predict.data_path" placeholder="/path/to/predict/data" />
         </div>
         <div class="field">
-          <label>{{ t('Source Data Format', 'Source Data Format') }}</label>
+          <label>{{ t('源数据格式', 'Source Data Format') }}</label>
           <select v-model="form.predict.source_data_format">
             <option value="parquet">parquet</option>
             <option value="csv">csv</option>
@@ -41,52 +41,56 @@
           </select>
         </div>
         <div class="field">
-          <label>{{ t('Output Name', 'Output Name') }}</label>
+          <label>{{ t('ID 列', 'ID Column') }}</label>
+          <input v-model="form.predict.id_column" placeholder="user_id" />
+        </div>
+        <div class="field">
+          <label>{{ t('输出名称', 'Output Name') }}</label>
           <input v-model="form.predict.name" placeholder="pred" />
         </div>
         <div class="field">
-          <label>{{ t('Save Data Format', 'Save Data Format') }}</label>
+          <label>{{ t('保存格式', 'Save Data Format') }}</label>
           <select v-model="form.predict.save_data_format">
             <option value="csv">csv</option>
             <option value="parquet">parquet</option>
           </select>
         </div>
         <div class="field">
-          <label>{{ t('Preview Rows', 'Preview Rows') }}</label>
+          <label>{{ t('预览行数', 'Preview Rows') }}</label>
           <input v-model.number="form.predict.preview_rows" type="number" />
         </div>
         <div class="field">
-          <label>{{ t('Batch Size', 'Batch Size') }}</label>
+          <label>{{ t('批大小', 'Batch Size') }}</label>
           <input v-model.number="form.predict.batch_size" type="number" />
         </div>
         <div class="field">
-          <label>{{ t('Num Workers', 'Num Workers') }}</label>
+          <label>{{ t('工作进程数', 'Num Workers') }}</label>
           <input v-model.number="form.predict.num_workers" type="number" />
         </div>
         <div class="field">
-          <label>{{ t('Num Processes', 'Num Processes') }}</label>
+          <label>{{ t('推理进程数', 'Num Processes') }}</label>
           <input v-model.number="form.predict.num_processes" type="number" />
         </div>
         <div class="field">
-          <label>{{ t('Device', 'Device') }}</label>
+          <label>{{ t('设备', 'Device') }}</label>
           <input v-model="form.predict.device" placeholder="cpu" />
         </div>
         <div class="field">
-          <label>{{ t('Use ONNX', 'Use ONNX') }}</label>
+          <label>{{ t('使用 ONNX', 'Use ONNX') }}</label>
           <select v-model="form.predict.use_onnx">
             <option :value="false">false</option>
             <option :value="true">true</option>
           </select>
         </div>
         <div class="field">
-          <label>{{ t('Streaming', 'Streaming') }}</label>
+          <label>{{ t('流式处理', 'Streaming') }}</label>
           <select v-model="form.predict.streaming">
             <option :value="true">true</option>
             <option :value="false">false</option>
           </select>
         </div>
         <div class="field">
-          <label>{{ t('Chunk Size', 'Chunk Size') }}</label>
+          <label>{{ t('分块大小', 'Chunk Size') }}</label>
           <input v-model.number="form.predict.chunk_size" type="number" />
         </div>
       </div>
@@ -98,7 +102,7 @@
         <textarea class="mono" readonly :value="yamlText"></textarea>
       </div>
       <div class="actions spaced-top">
-        <button class="primary" @click="download">{{ t('下载 predict_config.yaml', 'Download predict_config.yaml') }}</button>
+        <button class="primary" @click="download">{{ t('下载', 'Download') }}</button>
       </div>
     </div>
   </section>