()
| 86 | |
| 87 | |
| 88 | def diffusion_model_defaults(): |
| 89 | return dict( |
| 90 | in_channels=12, |
| 91 | model_channels=64, |
| 92 | out_channels=12, |
| 93 | num_res_blocks=1, |
| 94 | dropout=0, |
| 95 | channel_mult="1,2", |
| 96 | use_checkpoint=False, |
| 97 | use_fp16=False, |
| 98 | use_scale_shift_norm=True, |
| 99 | ) |
| 100 | |
| 101 | |
| 102 | def train_args(): |
no outgoing calls
no test coverage detected