diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py
index 385d8db..3e7687a 100644
--- a/src/aiunn/finetune.py
+++ b/src/aiunn/finetune.py
@@ -392,7 +392,7 @@ class FineTuner:
         
 if __name__ == "__main__":
     # Load your model first
-    model = AIIABase.load("/root/vision/dataset/AIIA-base-512")
+    model = AIIABase.load("/root/vision/AIIA/AIIA-base-512")
 
     trainer = FineTuner(
         model=model,