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

Training details #55

Closed
fradif96 opened this issue Dec 8, 2023 · 5 comments
Closed

Training details #55

fradif96 opened this issue Dec 8, 2023 · 5 comments

Comments

@fradif96
Copy link

fradif96 commented Dec 8, 2023

Hello, thank you for the great work.
I would kindly ask you for some training details. In particular:

  1. How many/which type of GPUs did you use for training? How much did it take to train?
  2. Which is the exact SD version checkpoint that has been fine-tuned? It is known from the report that it is the SD 2 v-model (Sec. 2.5) but it would be helpful to have the link to the exact SD checkpoint from which the fine-tuning has been initiated.

Thanks in advance

@eliphatfs
Copy link
Collaborator

  1. 24 A100s for 2 days for phase 1 and 16 A100s for 3.5 days for phase 2 training.
  2. I think there is only one model called SD 2 and is an v-model: https://huggingface.co/stabilityai/stable-diffusion-2

@netswift2905
Copy link

Really awesome. Do you plan to release the training code?

@eliphatfs
Copy link
Collaborator

#10. We are planning to release controlnet training code.

@shinetzh
Copy link

shinetzh commented May 7, 2024

  1. 24 A100s for 2 days for phase 1 and 16 A100s for 3.5 days for phase 2 training.
  2. I think there is only one model called SD 2 and is an v-model: https://huggingface.co/stabilityai/stable-diffusion-2

great work! what batch size did you use for training, and how many epoch?

@shinetzh
Copy link

shinetzh commented May 7, 2024

and, the reference latent is scaled by a factor of 5 in your paper, but i did not find the factor 5 used in inference code.

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

4 participants