Description of Evaluation Configuration Parameters
-
model_name:
Type:str
Description: Name of the model to use. Options include"SD-v1-4","SD-V2","SD-V2-1", etc. required: False -
encoder_model_name_or_path
Description: Model name or path for the encoder. Type:str
Example:CompVis/stable-diffusion-v1-4 -
target_ckpt:
Type:str
Description: Path to the target checkpoint. - If empty, the script will load the default model weights.
-
If provided, it supports both Diffusers-format checkpoints (directory) and CompVis checkpoints (file ending with
.pt). For CompVis, use the checkpoint of the model saved as Diffuser format. -
save_path:
Type:str
Description: Directory where the generated images will be saved. -
prompts_path:
Type:str
Description: Path to the CSV file containing prompts, evaluation seeds, and case numbers.
Default:"data/prompts/visualization_example.csv" -
guidance_scale:
Type:float
Description: Parameter that controls the classifier-free guidance during generation.
Default:7.5 -
image_size:
Type:int
Description: Dimensions of the generated images (height and width).
Default:512 -
ddim_steps:
Type:int
Description: Number of denoising steps (used in the diffusion process).
Default:100 -
num_samples:
Type:int
Description: Number of samples generated for each prompt.
Default:1 -
from_case:
Type:int
Description: Minimum case number from which to start generating images.
Default:0 -
folder_suffix:
Type:str
Description: Suffix added to the output folder name for visualizations. -
prompt_path:
Type:str
Description: Path to the CSV file containing prompts for evaluation.
Example:"data/prompts/coco_10k.csv" -
devices:
Type:str
Description: Comma-separated list of device IDs to be used during evaluation.
Example:"0,0"