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"