Skip to content

Commit 3243535

Browse files
committed
changed config path
1 parent fc60b27 commit 3243535

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/maxdiffusion/tests/ltx_transformer_step_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,9 @@ def test_one_step_transformer(self):
103103
key = jax.random.PRNGKey(42)
104104
devices_array = create_device_mesh(config)
105105
mesh = Mesh(devices_array, config.mesh_axes)
106-
config_path = "../models/ltx_video/xora_v1.2-13B-balanced-128.json"
106+
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+
107109
with open(config_path, "r") as f:
108110
model_config = json.load(f)
109111
relative_ckpt_path = model_config["ckpt_path"]

0 commit comments

Comments
 (0)