changed path

Author: Serenagu525

src/maxdiffusion/tests/ltx_transformer_step_test.py

@@ -39,7 +39,8 @@
 
 
 def load_ref_prediction():
-  saved_prediction_path = "../ltx_vid_transformer_test_ref_pred"
+  base_dir = os.path.dirname(__file__)
+  saved_prediction_path = os.path.join(base_dir, "ltx_vid_transformer_test_ref_pred")
   predict_dict = torch.load(saved_prediction_path)
   noise_pred_pt = predict_dict["noise_pred"].to(torch.float32)
   return noise_pred_pt