explainingai-code / StableDiffusion-PyTorch

This repo implements a Stable Diffusion model in PyTorch with all the essential components.
77 stars 15 forks source link

Removed hardcoded num_samples in ddpm #2

Closed kajc10 closed 4 months ago

kajc10 commented 4 months ago

Use train_config['num_samples'], not hardcoded 2.