[package] name = "comfyui-demo" version = "0.1.0" edition = "2021" [dependencies] clap = { version = "4.0", features = ["derive"] } serde = { version = "1.0", features = ["derive"] } serde_json = "1.0" anyhow = "1.0" stablediffusion = { path = "./crates/stable-diffusion-burn" } burn = "0.14.0" burn-autodiff = "0.14.0" burn-ndarray = "0.14.0" burn-wgpu = { version = "0.14.0", optional = true } # Only include burn-tch when needed for CPU/CUDA/MPS (not for Vulkan) # For Vulkan, we can use burn-wgpu directly without PyTorch image = "0.24.6" tokio = { version = "1.0", features = ["full"] } cfg-if = "1.0" [features] wgpu-backend = ["burn-wgpu"] [[bin]] name = "comfyui-cli" path = "src/main.rs"