v1.0.2; change high speed move, remove useless comments
This commit is contained in:
@@ -243,7 +243,7 @@ if __name__ == "__main__":
|
||||
|
||||
# 处理 MoE 和 绘图
|
||||
if isinstance(result, tuple):
|
||||
action, weights = result # moe
|
||||
action, (weights, latent) = result # moe
|
||||
action = action.detach().numpy().squeeze()[idx_model2mj]
|
||||
weights = weights.detach().numpy().squeeze()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user