Wan Demo throws No module named 'videox_fun.dist'

[jelling]

Are there any special installation instructions for using demo_wan.py?

When I run:

python demo_wanfun.py \
    --prompt "" \
    --checkpoint_path /workspace/ComfyUI/models/diffusion_models \
    --output_dir out \
    --input_path assets/videos/car_static.gif \
    --camera_motion "rot y 25" \
    --tracking_method spatracker \
    --override_extrinsics override \
    --gpu 0

I get:

 No module named 'videox_fun.dist'

Full Log

ARNING:py.warnings:/workspace/DiffusionAsShader/models/spatracker/models/core/spatracker/softsplat.py:285: FutureWarning: `torch.cuda.amp.custom_fwd(args...)` is deprecated. Please use `torch.amp.custom_fwd(args..., device_type='cuda')` instead.
  @torch.cuda.amp.custom_fwd(cast_inputs=torch.float32)

WARNING:py.warnings:/workspace/DiffusionAsShader/models/spatracker/models/core/spatracker/softsplat.py:368: FutureWarning: `torch.cuda.amp.custom_bwd(args...)` is deprecated. Please use `torch.amp.custom_bwd(args..., device_type='cuda')` instead.
  def backward(self, tenOutgrad):

Image input detected, using MoGe for tracking video generation.
WARNING:py.warnings:/workspace/DiffusionAsShader/submodules/MoGe/moge/model/v1.py:171: UserWarning: The following deprecated/invalid arguments are ignored: {'output_mask': True, 'split_head': True}
  warnings.warn(f"The following deprecated/invalid arguments are ignored: {deprecated_kwargs}")

Camera motion applied
WARNING:py.warnings:/workspace/DiffusionAsShader/models/pipelines_wanfun.py:422: RuntimeWarning: invalid value encountered in cast
  colors[:, :, 2] = (normalized_z * 255).astype(np.uint8)

rendering frames:   0%|                                                                                                                                                                                                                                     | 0/49 [00:00<?, ?it/s]WARNING:py.warnings:/workspace/DiffusionAsShader/models/pipelines_wanfun.py:435: RuntimeWarning: invalid value encountered in cast
  pixels = pixels.astype(int)

rendering frames: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 49/49 [01:06<00:00,  1.35s/it]
Video saved to out/tracking_video_moge.mp4
export tracking video via MoGe.
Traceback (most recent call last):
  File "/workspace/DiffusionAsShader/demo_wanfun.py", line 274, in <module>
    das.apply_tracking(
  File "/workspace/DiffusionAsShader/models/pipelines_wanfun.py", line 666, in apply_tracking
    self._infer_wanfun_ctrl(
  File "/usr/local/lib/python3.10/dist-packages/torch/utils/_contextlib.py", line 116, in decorate_context
    return func(*args, **kwargs)
  File "/workspace/DiffusionAsShader/models/pipelines_wanfun.py", line 97, in _infer_wanfun_ctrl
    from videox_fun.pipeline import WanFunControlPipeline
  File "/workspace/DiffusionAsShader/videox_fun/pipeline/__init__.py", line 1, in <module>
    from .pipeline_cogvideox_fun import CogVideoXFunPipeline
  File "/workspace/DiffusionAsShader/videox_fun/pipeline/pipeline_cogvideox_fun.py", line 31, in <module>
    from ..models import (AutoencoderKLCogVideoX,
  File "/workspace/DiffusionAsShader/videox_fun/models/__init__.py", line 3, in <module>
    from .cogvideox_transformer3d import CogVideoXTransformer3DModel
  File "/workspace/DiffusionAsShader/videox_fun/models/cogvideox_transformer3d.py", line 38, in <module>
    from ..dist import (get_sequence_parallel_rank,
ModuleNotFoundError: No module named 'videox_fun.dist'

[jelling]

I looked into this some more and it appears that 1) the dist directory from the VideoX repo is being used but 2) that directory has also been modified a bit.

Could someone please upload the correct dist directory? I tried patching things together but ran into additional issues.