mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-11 03:24:30 +00:00
refactor: unify model source and weight lifecycle management (#1956)
This commit is contained in:
1 parent
d04e8950c1
commit
6b47fec013
36 files changed
+2548
-1462
No files matched your search
@@ -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__
|
||||
Reference in new issue
Block a user