From 3d48dbdd1720715c1da9d1dbf0c509ee16e56a67 Mon Sep 17 00:00:00 2001
From: kaiza_hikaru <kaiza_hikaru@163.com>
Date: Mon, 3 Nov 2025 10:33:16 +0800
Subject: [PATCH] fix test

---
 MOAFTest.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/MOAFTest.py b/MOAFTest.py
index a0761fa..bbb6c00 100644
--- a/MOAFTest.py
+++ b/MOAFTest.py
@@ -73,7 +73,7 @@ def main():
         fusion_depth_list = [int(ch) for ch in model_type[2:]]
         model = MOAFWithSE(fusion_depth_list).to(device)
     elif "mmlp" in model_type:
-        model = MOAFWithMMLP(fusion_depth_list).to(device)
+        model = MOAFWithMMLP().to(device)
     else:
         model = MOAFNoFusion().to(device)