CYY

GPU memory used for stable diffusion

4

I finetune the stable diffusion on one RTX3090 with config ``` python main.py --logdir /tmp -t --postfix test -b configs/train_colossalai.yaml ``` The config for lighting is as follows:  ...

Good work! How to get patch embedding of image？

1

Excellent work! I want to know how to get image patch embedding, like output_hidden_states=True in CLIPVisionModelWithProjection. Thank you!