101 lines
2.8 KiB
JSON
101 lines
2.8 KiB
JSON
{
|
|
"seed": 12345,
|
|
"basesr": {
|
|
"sf": 4,
|
|
"chopping": {
|
|
"pch_size": 128,
|
|
"weight_type": "Gaussian",
|
|
"extra_bs": 8
|
|
}
|
|
},
|
|
"tiled_vae": true,
|
|
"latent_tiled_size": 128,
|
|
"sample_tiled_size": 1024,
|
|
"gradient_checkpointing_vae": true,
|
|
"sliced_vae": false,
|
|
"cfg_scale": 1.0,
|
|
"start_timesteps": 200,
|
|
"color_fix": "",
|
|
"base_model": "sd-turbo",
|
|
"sd_pipe": {
|
|
"target": "diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline",
|
|
"enable_grad_checkpoint": true,
|
|
"params": {
|
|
"pretrained_model_name_or_path": "stabilityai/sd-turbo",
|
|
"cache_dir": "./weights",
|
|
"use_safetensors": true,
|
|
"torch_dtype": "torch.float16"
|
|
}
|
|
},
|
|
"model_start": {
|
|
"target": "diffusers.models.autoencoders.NoisePredictor",
|
|
"ckpt_path": "weights/noise_predictor_sd_turbo_v5.pth",
|
|
"params": {
|
|
"in_channels": 3,
|
|
"down_block_types": [
|
|
"AttnDownBlock2D",
|
|
"AttnDownBlock2D"
|
|
],
|
|
"up_block_types": [
|
|
"AttnUpBlock2D",
|
|
"AttnUpBlock2D"
|
|
],
|
|
"block_out_channels": [
|
|
256,
|
|
512
|
|
],
|
|
"layers_per_block": [
|
|
3,
|
|
3
|
|
],
|
|
"act_fn": "silu",
|
|
"latent_channels": 4,
|
|
"norm_num_groups": 32,
|
|
"sample_size": 128,
|
|
"mid_block_add_attention": true,
|
|
"resnet_time_scale_shift": "default",
|
|
"temb_channels": 512,
|
|
"attention_head_dim": 64,
|
|
"freq_shift": 0,
|
|
"flip_sin_to_cos": true,
|
|
"double_z": true
|
|
}
|
|
},
|
|
"model_middle": {
|
|
"target": "diffusers.models.autoencoders.NoisePredictor",
|
|
"params": {
|
|
"in_channels": 3,
|
|
"down_block_types": [
|
|
"AttnDownBlock2D",
|
|
"AttnDownBlock2D"
|
|
],
|
|
"up_block_types": [
|
|
"AttnUpBlock2D",
|
|
"AttnUpBlock2D"
|
|
],
|
|
"block_out_channels": [
|
|
256,
|
|
512
|
|
],
|
|
"layers_per_block": [
|
|
3,
|
|
3
|
|
],
|
|
"act_fn": "silu",
|
|
"latent_channels": 4,
|
|
"norm_num_groups": 32,
|
|
"sample_size": 128,
|
|
"mid_block_add_attention": true,
|
|
"resnet_time_scale_shift": "default",
|
|
"temb_channels": 512,
|
|
"attention_head_dim": 64,
|
|
"freq_shift": 0,
|
|
"flip_sin_to_cos": true,
|
|
"double_z": true
|
|
}
|
|
},
|
|
"timesteps": [
|
|
200
|
|
],
|
|
"bs": 1
|
|
} |