We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fc60b27 commit 3243535Copy full SHA for 3243535
1 file changed
src/maxdiffusion/tests/ltx_transformer_step_test.py
@@ -103,7 +103,9 @@ def test_one_step_transformer(self):
103
key = jax.random.PRNGKey(42)
104
devices_array = create_device_mesh(config)
105
mesh = Mesh(devices_array, config.mesh_axes)
106
- config_path = "../models/ltx_video/xora_v1.2-13B-balanced-128.json"
+ base_dir = os.path.dirname(__file__)
107
+ config_path = os.path.join(base_dir, "../models/ltx_video/xora_v1.2-13B-balanced-128.json")
108
+
109
with open(config_path, "r") as f:
110
model_config = json.load(f)
111
relative_ckpt_path = model_config["ckpt_path"]
0 commit comments