refactor: unify model source and weight lifecycle management (#1956)

This commit is contained in:
leejet authored and GitHub committed 2026-09-10 23:53:48 +08:00
1 parent d04e8950c1
commit 6b47fec013
36 files changed
+2548 -1462

No files matched your search

+61
View File
@@ -0,0 +1,61 @@
#ifndef __SD_MODEL_COMPONENT_H__
#define __SD_MODEL_COMPONENT_H__
enum class ModelComponent {
Conditioner,
Diffusion,
HighNoiseDiffusion,
CLIPVision,
IPAdapter,
VAE,
PreviewVAE,
AudioVAE,
ControlNet,
PhotoMaker,
PuLID,
LoRA,
Upscaler,
Detector,
LatentUpsampler,
Count,
};
inline const char* model_component_name(ModelComponent component) {
switch (component) {
case ModelComponent::Conditioner:
return "Conditioner model";
case ModelComponent::Diffusion:
return "Diffusion model";
case ModelComponent::HighNoiseDiffusion:
return "High noise diffusion model";
case ModelComponent::CLIPVision:
return "CLIP vision";
case ModelComponent::IPAdapter:
return "IP-Adapter";
case ModelComponent::VAE:
return "VAE";
case ModelComponent::PreviewVAE:
return "preview VAE";
case ModelComponent::AudioVAE:
return "audio VAE";
case ModelComponent::ControlNet:
return "ControlNet";
case ModelComponent::PhotoMaker:
return "photomaker";
case ModelComponent::PuLID:
return "pulid";
case ModelComponent::LoRA:
return "LoRA";
case ModelComponent::Upscaler:
return "ESRGAN";
case ModelComponent::Detector:
return "YOLOv8";
case ModelComponent::LatentUpsampler:
return "LTX latent upsampler";
case ModelComponent::Count:
break;
}
return "unknown";
}
#endif // __SD_MODEL_COMPONENT_H__