From 81c66d406fbfd2bae5efc2bc338071161ce2e910 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Thu, 13 Mar 2025 23:27:54 -0400 Subject: [PATCH] don't download sdxl weights --- crates/desktop/src-tauri/src/ml/weights_registry/weights.rs | 2 ++ crates/desktop/src-tauri/src/threads/downloader_thread.rs | 6 +++--- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/crates/desktop/src-tauri/src/ml/weights_registry/weights.rs b/crates/desktop/src-tauri/src/ml/weights_registry/weights.rs index 332082eb40..4e44e453fa 100644 --- a/crates/desktop/src-tauri/src/ml/weights_registry/weights.rs +++ b/crates/desktop/src-tauri/src/ml/weights_registry/weights.rs @@ -42,12 +42,14 @@ pub const SDXL_TURBO_CLIP_TEXT_ENCODER_2 : WeightDescriptor = weight!( WeightFunction::TextEncoder ); +#[deprecated] pub const SDXL_TURBO_VAE : WeightDescriptor = weight!( "SDXL Turbo", "diffusion_pytorch_model.vae.safetensors", WeightFunction::Vae ); +#[deprecated] pub const SDXL_TURBO_UNET : WeightDescriptor = weight!( "SDXL Turbo", "diffusion_pytorch_model.unet.safetensors", diff --git a/crates/desktop/src-tauri/src/threads/downloader_thread.rs b/crates/desktop/src-tauri/src/threads/downloader_thread.rs index 5cb8046970..3a9992fc7d 100644 --- a/crates/desktop/src-tauri/src/threads/downloader_thread.rs +++ b/crates/desktop/src-tauri/src/threads/downloader_thread.rs @@ -45,9 +45,9 @@ pub async fn downloader_thread(app_data_root: AppDataRoot, app: AppHandle) -> ! download_queue.push_back(LYKON_DEAMSHAPER_7_VAE); download_queue.push_back(SIMIANLUO_LCM_DREAMSHAPER_V7_UNET); download_queue.push_back(SDXL_TURBO_CLIP_TEXT_ENCODER); // TODO(bt): Why is this still needed? - download_queue.push_back(SDXL_TURBO_CLIP_TEXT_ENCODER_2); // TODO(bt): REMOVE - download_queue.push_back(SDXL_TURBO_VAE); // TODO(bt): REMOVE - download_queue.push_back(SDXL_TURBO_UNET); // TODO(bt): REMOVE + //download_queue.push_back(SDXL_TURBO_CLIP_TEXT_ENCODER_2); // TODO(bt): REMOVE + //download_queue.push_back(SDXL_TURBO_VAE); // TODO(bt): REMOVE + //download_queue.push_back(SDXL_TURBO_UNET); // TODO(bt): REMOVE let mut handles = FuturesUnordered::new();