Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

wont start train #31

Open
theaorora opened this issue Dec 23, 2024 · 1 comment
Open

wont start train #31

theaorora opened this issue Dec 23, 2024 · 1 comment

Comments

@theaorora
Copy link

theaorora commented Dec 23, 2024

Hi tdrussell,
I followed the instruction but got this error

[2024-12-23 05:32:43,472] [INFO] [real_accelerator.py:222:get_accelerator] Setting ds_accelerator to cuda (auto detect)
[2024-12-23 05:32:47,733] [INFO] [launch.py:351:main] Process 3676 exits successfully.
(diffusion-pipe) root@5c5c3dc5ac6c:/workspace/diffusion-pipe#

my datasets is only images without caption.txt is this related to the problem? (i train person with flux before without .txt and no problem at all)
if i want to use video in datasets what format should i put? mp4?
can i mix images and videos in datasets?

@tdrussell
Copy link
Owner

I haven't seen this before. It looks like something is going wrong with deepspeed, it doesn't even launch the script, just immediately exits. What is the deepspeed command you're using to run it?

For video, I have only tested with mp4 but anything that the imageio library can load should work, which I think is most types of video file. You can put images and video in the same folder. But it may be convenient to separate them and have separate [[directory]] entries in the dataset config file.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants