separated checkpointer

Author: prishajain1

src/maxdiffusion/checkpointing/wan_checkpointer_2_1.py

@@ -3,7 +3,7 @@
 import jax
 import numpy as np
 from typing import Optional, Tuple
-from checkpointing.wan_checkpointer import WanCheckpointer
+from wan_checkpointer import WanCheckpointer
 from ..pipelines.wan.wan_pipeline import WanPipeline2_1
 from .. import max_logging
 import orbax.checkpoint as ocp

src/maxdiffusion/checkpointing/wan_checkpointer_2_2.py

@@ -7,7 +7,7 @@
 from .. import max_logging
 import orbax.checkpoint as ocp
 from etils import epath
-from checkpointing.wan_checkpointer import WanCheckpointer
+from wan_checkpointer import WanCheckpointer
 
 class WanCheckpointer2_2(WanCheckpointer):