kohya_ss
kohya_ss copied to clipboard
ValueError: num_samples should be a positive integer value, but got num_samples=0
Traceback (most recent call last):
File "/workspace/kohya_ss/sd-scripts/train_db.py", line 529, in <module>
train(args)
File "/workspace/kohya_ss/sd-scripts/train_db.py", line 190, in train
train_dataloader = torch.utils.data.DataLoader(
File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py", line 350, in __init__
sampler = RandomSampler(dataset, generator=generator) # type: ignore[arg-type]
File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/sampler.py", line 143, in __init__
raise ValueError(f"num_samples should be a positive integer value, but got num_samples={self.num_samples}")
ValueError: num_samples should be a positive integer value, but got num_samples=0
Traceback (most recent call last):
File "/workspace/kohya_ss/venv/bin/accelerate", line 8, in <module>
sys.exit(main())
File "/workspace/kohya_ss/venv/lib/python3.10/site-packages/accelerate/commands/accelerate_cli.py", line 47, in main
args.func(args)
File "/workspace/kohya_ss/venv/lib/python3.10/site-packages/accelerate/commands/launch.py", line 1017, in launch_command
simple_launcher(args)
File "/workspace/kohya_ss/venv/lib/python3.10/site-packages/accelerate/commands/launch.py", line 637, in simple_launcher
raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
subprocess.CalledProcessError: Command '['/workspace/kohya_ss/venv/bin/python3', '/workspace/kohya_ss/sd-scripts/train_db.py', '--config_file', '/workspace/kohya_ss/outputs/model/config_dreambooth-20240721-213429.toml']' returned non-zero exit status 1.
Not sure what I am doing wrong, worked before...
This happens for both Lora and Model training.