mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-09-24 20:20:37 +00:00
65 lines
1.7 KiB
C
65 lines
1.7 KiB
C
#ifndef __SD_MODEL_COMPONENT_H__
|
|
#define __SD_MODEL_COMPONENT_H__
|
|
|
|
enum class ModelComponent {
|
|
Conditioner,
|
|
Diffusion,
|
|
HighNoiseDiffusion,
|
|
CLIPVision,
|
|
IPAdapter,
|
|
VAE,
|
|
PreviewVAE,
|
|
AudioVAE,
|
|
AudioEncoder,
|
|
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::AudioEncoder:
|
|
return "audio encoder";
|
|
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__
|