(
image_size,
classifier_use_fp16,
classifier_width,
classifier_depth,
classifier_attention_resolutions,
classifier_use_scale_shift_norm,
classifier_resblock_updown,
classifier_pool,
learn_sigma,
diffusion_steps,
noise_schedule,
timestep_respacing,
use_kl,
predict_xstart,
rescale_timesteps,
rescale_learned_sigmas,
)
| 185 | |
| 186 | |
| 187 | def create_classifier_and_diffusion( |
| 188 | image_size, |
| 189 | classifier_use_fp16, |
| 190 | classifier_width, |
| 191 | classifier_depth, |
| 192 | classifier_attention_resolutions, |
| 193 | classifier_use_scale_shift_norm, |
| 194 | classifier_resblock_updown, |
| 195 | classifier_pool, |
| 196 | learn_sigma, |
| 197 | diffusion_steps, |
| 198 | noise_schedule, |
| 199 | timestep_respacing, |
| 200 | use_kl, |
| 201 | predict_xstart, |
| 202 | rescale_timesteps, |
| 203 | rescale_learned_sigmas, |
| 204 | ): |
| 205 | classifier = create_classifier( |
| 206 | image_size, |
| 207 | classifier_use_fp16, |
| 208 | classifier_width, |
| 209 | classifier_depth, |
| 210 | classifier_attention_resolutions, |
| 211 | classifier_use_scale_shift_norm, |
| 212 | classifier_resblock_updown, |
| 213 | classifier_pool, |
| 214 | ) |
| 215 | diffusion = create_gaussian_diffusion( |
| 216 | steps=diffusion_steps, |
| 217 | learn_sigma=learn_sigma, |
| 218 | noise_schedule=noise_schedule, |
| 219 | use_kl=use_kl, |
| 220 | predict_xstart=predict_xstart, |
| 221 | rescale_timesteps=rescale_timesteps, |
| 222 | rescale_learned_sigmas=rescale_learned_sigmas, |
| 223 | timestep_respacing=timestep_respacing, |
| 224 | ) |
| 225 | return classifier, diffusion |
| 226 | |
| 227 | |
| 228 | def create_classifier( |
no test coverage detected