#include <scheduler.h>

Inheritance diagram for nndeploy::stable_diffusion::SchedulerParam:

Collaboration diagram for nndeploy::stable_diffusion::SchedulerParam:

Public Member Functions
virtual base::Status	deserialize (rapidjson::Value &json)

SchedulerParam &	operator= (const SchedulerParam &param)

	PARAM_COPY (SchedulerParam)

	PARAM_COPY_TO (SchedulerParam)

	SchedulerParam ()

virtual base::Status	serialize (rapidjson::Value &json, rapidjson::Document::AllocatorType &allocator)

virtual	~SchedulerParam ()

Public Member Functions inherited from nndeploy::base::Param
base::Status	addDropdownParam (const std::string &dropdown_param, const std::vector< std::string > &dropdown_values)

base::Status	addIoParam (const std::string &io_param)

base::Status	addRequiredParam (const std::string &required_param)

base::Status	addUiParam (const std::string &ui_param)

base::Status	clearDropdownParams ()

base::Status	clearIoParams ()

base::Status	clearRequiredParams ()

base::Status	clearUiParams ()

virtual base::Status	deserialize (const std::string &json_str)

virtual base::Status	get (const std::string &key, base::Any &any)

std::map< std::string, std::vector< std::string > >	getDropdownParams ()

std::vector< std::string >	getIoParams ()

std::vector< std::string >	getRequiredParams ()

std::vector< std::string >	getUiParams ()

virtual base::Status	loadFile (const std::string &path)

	Param ()

base::Status	removeDropdownParam (const std::string &dropdown_param)

base::Status	removeIoParam (const std::string &io_param)

base::Status	removeRequiredParam (const std::string &required_param)

base::Status	removeUiParam (const std::string &ui_param)

virtual base::Status	saveFile (const std::string &path)

virtual std::string	serialize ()

virtual base::Status	set (const std::string &key, base::Any &any)

base::Status	setDropdownParams (const std::map< std::string, std::vector< std::string >> &dropdown_params)

base::Status	setIoParams (const std::vector< std::string > &io_params)

base::Status	setRequiredParams (const std::vector< std::string > &required_params)

base::Status	setUiParams (const std::vector< std::string > &ui_params)

virtual	~Param ()

Public Attributes
bool	clip_sample_ = false

float	guidance_scale_ = 7.5

int	image_height_ = 512

int	image_width_ = 512

float	init_noise_sigma_ = 1.0f

int	num_inference_steps_ = 50

int	num_train_timesteps_ = 1000

int	unet_channels_ = 4

float	vae_scale_factor_ = 0.18215

std::string	version_ = "v1.5"

Public Attributes inherited from nndeploy::base::Param
std::map< std::string, std::vector< std::string > >	dropdown_params_

std::vector< std::string >	io_params_

std::vector< std::string >	required_params_

std::vector< std::string >	ui_params_

Detailed Description

Definition at line 30 of file scheduler.h.

Constructor & Destructor Documentation

◆ SchedulerParam()

nndeploy::stable_diffusion::SchedulerParam::SchedulerParam ( )

inline

Definition at line 32 of file scheduler.h.

◆ ~SchedulerParam()

virtual nndeploy::stable_diffusion::SchedulerParam::~SchedulerParam ( )

inlinevirtual

Definition at line 33 of file scheduler.h.

Member Function Documentation

◆ deserialize()

virtual base::Status nndeploy::stable_diffusion::SchedulerParam::deserialize ( rapidjson::Value & json )

virtual

Reimplemented from nndeploy::base::Param.

Reimplemented in nndeploy::stable_diffusion::DDIMSchedulerParam.

◆ operator=()

SchedulerParam& nndeploy::stable_diffusion::SchedulerParam::operator= ( const SchedulerParam & param )

inline

Definition at line 38 of file scheduler.h.

◆ PARAM_COPY()

nndeploy::stable_diffusion::SchedulerParam::PARAM_COPY ( SchedulerParam )

◆ PARAM_COPY_TO()

nndeploy::stable_diffusion::SchedulerParam::PARAM_COPY_TO ( SchedulerParam )

◆ serialize()

virtual base::Status nndeploy::stable_diffusion::SchedulerParam::serialize	(	rapidjson::Value &	json,
		rapidjson::Document::AllocatorType &	allocator
	)

virtual

Reimplemented from nndeploy::base::Param.

Reimplemented in nndeploy::stable_diffusion::DDIMSchedulerParam.

Member Data Documentation

◆ clip_sample_

bool nndeploy::stable_diffusion::SchedulerParam::clip_sample_ = false

Definition at line 56 of file scheduler.h.

◆ guidance_scale_

float nndeploy::stable_diffusion::SchedulerParam::guidance_scale_ = 7.5

Definition at line 61 of file scheduler.h.

◆ image_height_

int nndeploy::stable_diffusion::SchedulerParam::image_height_ = 512

Definition at line 59 of file scheduler.h.

◆ image_width_

int nndeploy::stable_diffusion::SchedulerParam::image_width_ = 512

Definition at line 60 of file scheduler.h.

◆ init_noise_sigma_

float nndeploy::stable_diffusion::SchedulerParam::init_noise_sigma_ = 1.0f

Definition at line 63 of file scheduler.h.

◆ num_inference_steps_

int nndeploy::stable_diffusion::SchedulerParam::num_inference_steps_ = 50

Definition at line 57 of file scheduler.h.

◆ num_train_timesteps_

int nndeploy::stable_diffusion::SchedulerParam::num_train_timesteps_ = 1000

Definition at line 55 of file scheduler.h.

◆ unet_channels_

int nndeploy::stable_diffusion::SchedulerParam::unet_channels_ = 4

Definition at line 58 of file scheduler.h.

◆ vae_scale_factor_

float nndeploy::stable_diffusion::SchedulerParam::vae_scale_factor_ = 0.18215

Definition at line 62 of file scheduler.h.

◆ version_

std::string nndeploy::stable_diffusion::SchedulerParam::version_ = "v1.5"

Definition at line 54 of file scheduler.h.

The documentation for this class was generated from the following file:

/home/docs/checkouts/readthedocs.org/user_builds/nndeploy-zh/checkouts/latest/plugin/include/nndeploy/stable_diffusion/scheduler.h

Public Member Functions

Public Attributes