Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
db33af06
Unverified
Commit
db33af06
authored
May 24, 2024
by
Tolga Cangöz
Committed by
GitHub
May 24, 2024
Browse files
Fix a grammatical error in the `raise` messages (#8272)
Fix grammatical error
parent
1096f88e
Changes
27
Hide whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
9 additions
and
9 deletions
+9
-9
src/diffusers/schedulers/scheduling_lcm.py
src/diffusers/schedulers/scheduling_lcm.py
+1
-1
src/diffusers/schedulers/scheduling_lms_discrete.py
src/diffusers/schedulers/scheduling_lms_discrete.py
+1
-1
src/diffusers/schedulers/scheduling_pndm.py
src/diffusers/schedulers/scheduling_pndm.py
+1
-1
src/diffusers/schedulers/scheduling_repaint.py
src/diffusers/schedulers/scheduling_repaint.py
+1
-1
src/diffusers/schedulers/scheduling_sasolver.py
src/diffusers/schedulers/scheduling_sasolver.py
+2
-2
src/diffusers/schedulers/scheduling_tcd.py
src/diffusers/schedulers/scheduling_tcd.py
+1
-1
src/diffusers/schedulers/scheduling_unipc_multistep.py
src/diffusers/schedulers/scheduling_unipc_multistep.py
+2
-2
No files found.
src/diffusers/schedulers/scheduling_lcm.py
View file @
db33af06
...
@@ -224,7 +224,7 @@ class LCMScheduler(SchedulerMixin, ConfigMixin):
...
@@ -224,7 +224,7 @@ class LCMScheduler(SchedulerMixin, ConfigMixin):
# Glide cosine schedule
# Glide cosine schedule
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
else
:
else
:
raise
NotImplementedError
(
f
"
{
beta_schedule
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
beta_schedule
}
is not implemented for
{
self
.
__class__
}
"
)
# Rescale for zero SNR
# Rescale for zero SNR
if
rescale_betas_zero_snr
:
if
rescale_betas_zero_snr
:
...
...
src/diffusers/schedulers/scheduling_lms_discrete.py
View file @
db33af06
...
@@ -149,7 +149,7 @@ class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin):
...
@@ -149,7 +149,7 @@ class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin):
# Glide cosine schedule
# Glide cosine schedule
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
else
:
else
:
raise
NotImplementedError
(
f
"
{
beta_schedule
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
beta_schedule
}
is not implemented for
{
self
.
__class__
}
"
)
self
.
alphas
=
1.0
-
self
.
betas
self
.
alphas
=
1.0
-
self
.
betas
self
.
alphas_cumprod
=
torch
.
cumprod
(
self
.
alphas
,
dim
=
0
)
self
.
alphas_cumprod
=
torch
.
cumprod
(
self
.
alphas
,
dim
=
0
)
...
...
src/diffusers/schedulers/scheduling_pndm.py
View file @
db33af06
...
@@ -135,7 +135,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):
...
@@ -135,7 +135,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):
# Glide cosine schedule
# Glide cosine schedule
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
else
:
else
:
raise
NotImplementedError
(
f
"
{
beta_schedule
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
beta_schedule
}
is not implemented for
{
self
.
__class__
}
"
)
self
.
alphas
=
1.0
-
self
.
betas
self
.
alphas
=
1.0
-
self
.
betas
self
.
alphas_cumprod
=
torch
.
cumprod
(
self
.
alphas
,
dim
=
0
)
self
.
alphas_cumprod
=
torch
.
cumprod
(
self
.
alphas
,
dim
=
0
)
...
...
src/diffusers/schedulers/scheduling_repaint.py
View file @
db33af06
...
@@ -143,7 +143,7 @@ class RePaintScheduler(SchedulerMixin, ConfigMixin):
...
@@ -143,7 +143,7 @@ class RePaintScheduler(SchedulerMixin, ConfigMixin):
betas
=
torch
.
linspace
(
-
6
,
6
,
num_train_timesteps
)
betas
=
torch
.
linspace
(
-
6
,
6
,
num_train_timesteps
)
self
.
betas
=
torch
.
sigmoid
(
betas
)
*
(
beta_end
-
beta_start
)
+
beta_start
self
.
betas
=
torch
.
sigmoid
(
betas
)
*
(
beta_end
-
beta_start
)
+
beta_start
else
:
else
:
raise
NotImplementedError
(
f
"
{
beta_schedule
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
beta_schedule
}
is not implemented for
{
self
.
__class__
}
"
)
self
.
alphas
=
1.0
-
self
.
betas
self
.
alphas
=
1.0
-
self
.
betas
self
.
alphas_cumprod
=
torch
.
cumprod
(
self
.
alphas
,
dim
=
0
)
self
.
alphas_cumprod
=
torch
.
cumprod
(
self
.
alphas
,
dim
=
0
)
...
...
src/diffusers/schedulers/scheduling_sasolver.py
View file @
db33af06
...
@@ -180,7 +180,7 @@ class SASolverScheduler(SchedulerMixin, ConfigMixin):
...
@@ -180,7 +180,7 @@ class SASolverScheduler(SchedulerMixin, ConfigMixin):
# Glide cosine schedule
# Glide cosine schedule
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
else
:
else
:
raise
NotImplementedError
(
f
"
{
beta_schedule
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
beta_schedule
}
is not implemented for
{
self
.
__class__
}
"
)
self
.
alphas
=
1.0
-
self
.
betas
self
.
alphas
=
1.0
-
self
.
betas
self
.
alphas_cumprod
=
torch
.
cumprod
(
self
.
alphas
,
dim
=
0
)
self
.
alphas_cumprod
=
torch
.
cumprod
(
self
.
alphas
,
dim
=
0
)
...
@@ -194,7 +194,7 @@ class SASolverScheduler(SchedulerMixin, ConfigMixin):
...
@@ -194,7 +194,7 @@ class SASolverScheduler(SchedulerMixin, ConfigMixin):
self
.
init_noise_sigma
=
1.0
self
.
init_noise_sigma
=
1.0
if
algorithm_type
not
in
[
"data_prediction"
,
"noise_prediction"
]:
if
algorithm_type
not
in
[
"data_prediction"
,
"noise_prediction"
]:
raise
NotImplementedError
(
f
"
{
algorithm_type
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
algorithm_type
}
is not implemented for
{
self
.
__class__
}
"
)
# setable values
# setable values
self
.
num_inference_steps
=
None
self
.
num_inference_steps
=
None
...
...
src/diffusers/schedulers/scheduling_tcd.py
View file @
db33af06
...
@@ -225,7 +225,7 @@ class TCDScheduler(SchedulerMixin, ConfigMixin):
...
@@ -225,7 +225,7 @@ class TCDScheduler(SchedulerMixin, ConfigMixin):
# Glide cosine schedule
# Glide cosine schedule
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
else
:
else
:
raise
NotImplementedError
(
f
"
{
beta_schedule
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
beta_schedule
}
is not implemented for
{
self
.
__class__
}
"
)
# Rescale for zero SNR
# Rescale for zero SNR
if
rescale_betas_zero_snr
:
if
rescale_betas_zero_snr
:
...
...
src/diffusers/schedulers/scheduling_unipc_multistep.py
View file @
db33af06
...
@@ -211,7 +211,7 @@ class UniPCMultistepScheduler(SchedulerMixin, ConfigMixin):
...
@@ -211,7 +211,7 @@ class UniPCMultistepScheduler(SchedulerMixin, ConfigMixin):
# Glide cosine schedule
# Glide cosine schedule
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
self
.
betas
=
betas_for_alpha_bar
(
num_train_timesteps
)
else
:
else
:
raise
NotImplementedError
(
f
"
{
beta_schedule
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
beta_schedule
}
is not implemented for
{
self
.
__class__
}
"
)
if
rescale_betas_zero_snr
:
if
rescale_betas_zero_snr
:
self
.
betas
=
rescale_zero_terminal_snr
(
self
.
betas
)
self
.
betas
=
rescale_zero_terminal_snr
(
self
.
betas
)
...
@@ -237,7 +237,7 @@ class UniPCMultistepScheduler(SchedulerMixin, ConfigMixin):
...
@@ -237,7 +237,7 @@ class UniPCMultistepScheduler(SchedulerMixin, ConfigMixin):
if
solver_type
in
[
"midpoint"
,
"heun"
,
"logrho"
]:
if
solver_type
in
[
"midpoint"
,
"heun"
,
"logrho"
]:
self
.
register_to_config
(
solver_type
=
"bh2"
)
self
.
register_to_config
(
solver_type
=
"bh2"
)
else
:
else
:
raise
NotImplementedError
(
f
"
{
solver_type
}
does
is not implemented for
{
self
.
__class__
}
"
)
raise
NotImplementedError
(
f
"
{
solver_type
}
is not implemented for
{
self
.
__class__
}
"
)
self
.
predict_x0
=
predict_x0
self
.
predict_x0
=
predict_x0
# setable values
# setable values
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment