diff options
Diffstat (limited to 'candle-examples/examples/stable-diffusion/main.rs')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/main.rs | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/candle-examples/examples/stable-diffusion/main.rs b/candle-examples/examples/stable-diffusion/main.rs index 31848f38..33a1dc71 100644 --- a/candle-examples/examples/stable-diffusion/main.rs +++ b/candle-examples/examples/stable-diffusion/main.rs @@ -3,8 +3,11 @@ extern crate intel_mkl_src; mod attention; mod clip; +mod ddim; mod embeddings; mod resnet; +mod schedulers; +mod stable_diffusion; mod unet_2d; mod unet_2d_blocks; mod utils; |