feature-wan-2-2 branch. It founds images but shows dataset error.
dataset file:
[general]
resolution = [960, 960]
caption_extension = ".txt"
batch_size = 1
enable_bucket = true
bucket_no_upscale = false
[[datasets]]
image_directory = "/workspace/musubi-tuner/dataset"
cache_directory = "/workspace/musubi-tuner/dataset/cache"
num_repeats = 1 # optional, default is 1. Number of times to repeat the dataset. Useful to balance the multiple datasets with different sizes.
Error itself:
Trying to import sageattention
Failed to import sageattention
INFO:musubi_tuner.wan.modules.model:Detected DiT dtype: torch.float16
INFO:musubi_tuner.hv_train_network:Load dataset config from /workspace/musubi-tuner/dataset/dataset.toml
INFO:musubi_tuner.dataset.image_video_dataset:glob images in /workspace/musubi-tuner/dataset
INFO:musubi_tuner.dataset.image_video_dataset:found 14 images
INFO:musubi_tuner.dataset.config_utils:[Dataset 0]
is_image_dataset: True
resolution: (960, 960)
batch_size: 1
num_repeats: 1
caption_extension: ".txt"
enable_bucket: True
bucket_no_upscale: True
cache_directory: "/workspace/musubi-tuner/dataset/cache"
debug_dataset: False
image_directory: "/workspace/musubi-tuner/dataset"
image_jsonl_file: "None"
fp_latent_window_size: 9
fp_1f_clean_indices: None
fp_1f_target_index: None
fp_1f_no_post: False
flux_kontext_no_resize_control: False
INFO:musubi_tuner.dataset.image_video_dataset:total batches: 0
Traceback (most recent call last):
File "/workspace/musubi-tuner/src/musubi_tuner/wan_train_network.py", line 551, in
main()
File "/workspace/musubi-tuner/src/musubi_tuner/wan_train_network.py", line 547, in main
trainer.train(args)
File "/workspace/musubi-tuner/src/musubi_tuner/hv_train_network.py", line 1488, in train
raise ValueError("No training items found in the dataset / データセットに学習データがありません")
ValueError: No training items found in the dataset / データセットに学習データがありません
Traceback (most recent call last):
File "/usr/local/bin/accelerate", line 8, in
sys.exit(main())
^^^^^^
File "/usr/local/lib/python3.11/dist-packages/accelerate/commands/accelerate_cli.py", line 50, in main
args.func(args)
File "/usr/local/lib/python3.11/dist-packages/accelerate/commands/launch.py", line 1213, in launch_command
simple_launcher(args)
File "/usr/local/lib/python3.11/dist-packages/accelerate/commands/launch.py", line 795, in simple_launcher
raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
feature-wan-2-2 branch. It founds images but shows dataset error.
dataset file:
Error itself: