File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -59,7 +59,7 @@ def __init__(
59
59
trained_betas = None ,
60
60
timestep_values = None ,
61
61
clip_sample = True ,
62
- clip_alpha_at_one = True ,
62
+ set_alpha_to_one = True ,
63
63
tensor_format = "pt" ,
64
64
):
65
65
@@ -79,9 +79,9 @@ def __init__(
79
79
80
80
# At every step in ddim, we are looking into the previous alphas_cumprod
81
81
# For the final step, there is no previous alphas_cumprod because we are already at 0
82
- # `clip_alpha_at_one ` decides whether we set this paratemer simply to one or
82
+ # `set_alpha_to_one ` decides whether we set this paratemer simply to one or
83
83
# whether we use the final alpha of the "non-previous" one.
84
- self .final_alpha_cumprod = np .array (1.0 ) if clip_alpha_at_one else self .alphas_cumprod [0 ]
84
+ self .final_alpha_cumprod = np .array (1.0 ) if set_alpha_to_one else self .alphas_cumprod [0 ]
85
85
86
86
# setable values
87
87
self .num_inference_steps = None
Original file line number Diff line number Diff line change @@ -871,7 +871,7 @@ def test_stable_diffusion_fast_ddim(self):
871
871
beta_end = 0.012 ,
872
872
beta_schedule = "scaled_linear" ,
873
873
clip_sample = False ,
874
- clip_alpha_at_one = False ,
874
+ set_alpha_to_one = False ,
875
875
)
876
876
sd_pipe .scheduler = scheduler
877
877
You can’t perform that action at this time.
0 commit comments