huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move the stable-diffusion modeling code so that it's easier to re-use. (#812)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-11 11:45:57 +01:00
committed by GitHub
parent 84ee870efd
commit d7b9fec849
13 changed files with 28 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
candle-examples/examples/stable-diffusion/main.rs
View File

@ -4,17 +4,7 @@ extern crate accelerate_src;
#[cfg(feature = "mkl")]
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;
mod vae;
use candle_transformers::models::stable_diffusion;
use anyhow::{Error as E, Result};
use candle::{DType, Device, IndexOp, Tensor, D};