This commit is contained in:
Yuwei Guo
2023-07-18 16:41:22 +08:00
parent b7bd475c1e
commit 4f0efbe11f
4 changed files with 346 additions and 3 deletions

View File

@@ -21,9 +21,6 @@ unet_additional_kwargs:
temporal_attention_dim_div: 1
noise_scheduler_kwargs:
num_train_timesteps: 1000
beta_start: 0.00085
beta_end: 0.012
beta_schedule: "linear"
steps_offset: 1
clip_sample: false