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
c50d9975
Commit
c50d9975
authored
Jun 30, 2022
by
patil-suraj
Browse files
remove unused args
parent
f1cb8074
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
14 deletions
+2
-14
src/diffusers/models/unet_sde_score_estimation.py
src/diffusers/models/unet_sde_score_estimation.py
+2
-14
No files found.
src/diffusers/models/unet_sde_score_estimation.py
View file @
c50d9975
...
@@ -396,10 +396,8 @@ class NCSNpp(ModelMixin, ConfigMixin):
...
@@ -396,10 +396,8 @@ class NCSNpp(ModelMixin, ConfigMixin):
def
__init__
(
def
__init__
(
self
,
self
,
centered
=
False
,
image_size
=
1024
,
image_size
=
1024
,
num_channels
=
3
,
num_channels
=
3
,
attention_type
=
"ddpm"
,
attn_resolutions
=
(
16
,),
attn_resolutions
=
(
16
,),
ch_mult
=
(
1
,
2
,
4
,
8
,
16
,
32
,
32
,
32
),
ch_mult
=
(
1
,
2
,
4
,
8
,
16
,
32
,
32
,
32
),
conditional
=
True
,
conditional
=
True
,
...
@@ -411,24 +409,19 @@ class NCSNpp(ModelMixin, ConfigMixin):
...
@@ -411,24 +409,19 @@ class NCSNpp(ModelMixin, ConfigMixin):
fourier_scale
=
16
,
fourier_scale
=
16
,
init_scale
=
0.0
,
init_scale
=
0.0
,
nf
=
16
,
nf
=
16
,
nonlinearity
=
"swish"
,
normalization
=
"GroupNorm"
,
num_res_blocks
=
1
,
num_res_blocks
=
1
,
progressive
=
"output_skip"
,
progressive
=
"output_skip"
,
progressive_combine
=
"sum"
,
progressive_combine
=
"sum"
,
progressive_input
=
"input_skip"
,
progressive_input
=
"input_skip"
,
resamp_with_conv
=
True
,
resamp_with_conv
=
True
,
resblock_type
=
"biggan"
,
scale_by_sigma
=
True
,
scale_by_sigma
=
True
,
skip_rescale
=
True
,
skip_rescale
=
True
,
continuous
=
True
,
continuous
=
True
,
):
):
super
().
__init__
()
super
().
__init__
()
self
.
register_to_config
(
self
.
register_to_config
(
centered
=
centered
,
image_size
=
image_size
,
image_size
=
image_size
,
num_channels
=
num_channels
,
num_channels
=
num_channels
,
attention_type
=
attention_type
,
attn_resolutions
=
attn_resolutions
,
attn_resolutions
=
attn_resolutions
,
ch_mult
=
ch_mult
,
ch_mult
=
ch_mult
,
conditional
=
conditional
,
conditional
=
conditional
,
...
@@ -440,14 +433,11 @@ class NCSNpp(ModelMixin, ConfigMixin):
...
@@ -440,14 +433,11 @@ class NCSNpp(ModelMixin, ConfigMixin):
fourier_scale
=
fourier_scale
,
fourier_scale
=
fourier_scale
,
init_scale
=
init_scale
,
init_scale
=
init_scale
,
nf
=
nf
,
nf
=
nf
,
nonlinearity
=
nonlinearity
,
normalization
=
normalization
,
num_res_blocks
=
num_res_blocks
,
num_res_blocks
=
num_res_blocks
,
progressive
=
progressive
,
progressive
=
progressive
,
progressive_combine
=
progressive_combine
,
progressive_combine
=
progressive_combine
,
progressive_input
=
progressive_input
,
progressive_input
=
progressive_input
,
resamp_with_conv
=
resamp_with_conv
,
resamp_with_conv
=
resamp_with_conv
,
resblock_type
=
resblock_type
,
scale_by_sigma
=
scale_by_sigma
,
scale_by_sigma
=
scale_by_sigma
,
skip_rescale
=
skip_rescale
,
skip_rescale
=
skip_rescale
,
continuous
=
continuous
,
continuous
=
continuous
,
...
@@ -462,7 +452,6 @@ class NCSNpp(ModelMixin, ConfigMixin):
...
@@ -462,7 +452,6 @@ class NCSNpp(ModelMixin, ConfigMixin):
self
.
conditional
=
conditional
self
.
conditional
=
conditional
self
.
skip_rescale
=
skip_rescale
self
.
skip_rescale
=
skip_rescale
self
.
resblock_type
=
resblock_type
self
.
progressive
=
progressive
self
.
progressive
=
progressive
self
.
progressive_input
=
progressive_input
self
.
progressive_input
=
progressive_input
self
.
embedding_type
=
embedding_type
self
.
embedding_type
=
embedding_type
...
@@ -633,7 +622,6 @@ class NCSNpp(ModelMixin, ConfigMixin):
...
@@ -633,7 +622,6 @@ class NCSNpp(ModelMixin, ConfigMixin):
else
:
else
:
temb
=
None
temb
=
None
if
not
self
.
config
.
centered
:
# If input data is in [0, 1]
# If input data is in [0, 1]
x
=
2
*
x
-
1.0
x
=
2
*
x
-
1.0
...
...
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