feat: 更新 README,添加生成长度控制和测试指令;修改 main_chat.py,优化模型加载和生成逻辑;新增单元测试文件 test_smoke.py

This commit is contained in:
2026-07-08 11:48:19 +08:00
parent 1de4656eb8
commit 24b56ebb6b
5 changed files with 177 additions and 20 deletions

View File

@@ -436,11 +436,13 @@ class MultimodalJSpaceModel(nn.Module):
x = self.encode_modality(modality, data)
# 处理序列或单步
w_traj_all = []
if x.dim() == 2: # 单步 (batch, input_dim)
state, w_traj = self.step(state, x, record_trajectory)
if record_trajectory:
w_traj_all = w_traj
w = state['w']
else: # 序列 (batch, T, input_dim)
w_traj_all = []
for t in range(x.shape[1]):
state, w_traj = self.step(state, x[:, t], record_trajectory)
if record_trajectory: