Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
diffusers
Commits
b2b3b1a8
Unverified
Commit
b2b3b1a8
authored
Sep 08, 2022
by
Patrick von Platen
Committed by
GitHub
Sep 08, 2022
Browse files
[Black] Update black (#433)
* Update black * update table
parent
44968e42
Changes
9
Hide whitespace changes
Inline
Side-by-side
Showing
9 changed files
with
3 additions
and
16 deletions
+3
-16
examples/textual_inversion/textual_inversion.py
examples/textual_inversion/textual_inversion.py
+0
-1
setup.py
setup.py
+2
-2
src/diffusers/dependency_versions_table.py
src/diffusers/dependency_versions_table.py
+1
-1
src/diffusers/models/unet_blocks.py
src/diffusers/models/unet_blocks.py
+0
-3
src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py
...s/pipelines/latent_diffusion/pipeline_latent_diffusion.py
+0
-1
src/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py
...tent_diffusion_uncond/pipeline_latent_diffusion_uncond.py
+0
-1
src/diffusers/schedulers/scheduling_ddpm.py
src/diffusers/schedulers/scheduling_ddpm.py
+0
-2
src/diffusers/schedulers/scheduling_sde_vp.py
src/diffusers/schedulers/scheduling_sde_vp.py
+0
-1
src/diffusers/utils/logging.py
src/diffusers/utils/logging.py
+0
-4
No files found.
examples/textual_inversion/textual_inversion.py
View file @
b2b3b1a8
...
...
@@ -238,7 +238,6 @@ class TextualInversionDataset(Dataset):
placeholder_token
=
"*"
,
center_crop
=
False
,
):
self
.
data_root
=
data_root
self
.
tokenizer
=
tokenizer
self
.
learnable_property
=
learnable_property
...
...
setup.py
View file @
b2b3b1a8
...
...
@@ -78,7 +78,7 @@ from setuptools import find_packages, setup
_deps
=
[
"Pillow"
,
"accelerate>=0.11.0"
,
"black==22.
3
"
,
"black==22.
8
"
,
"datasets"
,
"filelock"
,
"flake8>=3.8.3"
,
...
...
@@ -167,7 +167,7 @@ extras = {}
extras
=
{}
extras
[
"quality"
]
=
[
"black==22.
3
"
,
"isort>=5.5.4"
,
"flake8>=3.8.3"
,
"hf-doc-builder"
]
extras
[
"quality"
]
=
[
"black==22.
8
"
,
"isort>=5.5.4"
,
"flake8>=3.8.3"
,
"hf-doc-builder"
]
extras
[
"docs"
]
=
[
"hf-doc-builder"
]
extras
[
"training"
]
=
[
"accelerate"
,
"datasets"
,
"tensorboard"
,
"modelcards"
]
extras
[
"test"
]
=
[
"datasets"
,
"onnxruntime"
,
"pytest"
,
"pytest-timeout"
,
"pytest-xdist"
,
"scipy"
,
"transformers"
]
...
...
src/diffusers/dependency_versions_table.py
View file @
b2b3b1a8
...
...
@@ -4,7 +4,7 @@
deps
=
{
"Pillow"
:
"Pillow"
,
"accelerate"
:
"accelerate>=0.11.0"
,
"black"
:
"black==22.
3
"
,
"black"
:
"black==22.
8
"
,
"datasets"
:
"datasets"
,
"filelock"
:
"filelock"
,
"flake8"
:
"flake8>=3.8.3"
,
...
...
src/diffusers/models/unet_blocks.py
View file @
b2b3b1a8
...
...
@@ -979,7 +979,6 @@ class AttnUpBlock2D(nn.Module):
def
forward
(
self
,
hidden_states
,
res_hidden_states_tuple
,
temb
=
None
):
for
resnet
,
attn
in
zip
(
self
.
resnets
,
self
.
attentions
):
# pop res hidden states
res_hidden_states
=
res_hidden_states_tuple
[
-
1
]
res_hidden_states_tuple
=
res_hidden_states_tuple
[:
-
1
]
...
...
@@ -1075,7 +1074,6 @@ class CrossAttnUpBlock2D(nn.Module):
def
forward
(
self
,
hidden_states
,
res_hidden_states_tuple
,
temb
=
None
,
encoder_hidden_states
=
None
):
for
resnet
,
attn
in
zip
(
self
.
resnets
,
self
.
attentions
):
# pop res hidden states
res_hidden_states
=
res_hidden_states_tuple
[
-
1
]
res_hidden_states_tuple
=
res_hidden_states_tuple
[:
-
1
]
...
...
@@ -1139,7 +1137,6 @@ class UpBlock2D(nn.Module):
def
forward
(
self
,
hidden_states
,
res_hidden_states_tuple
,
temb
=
None
):
for
resnet
in
self
.
resnets
:
# pop res hidden states
res_hidden_states
=
res_hidden_states_tuple
[
-
1
]
res_hidden_states_tuple
=
res_hidden_states_tuple
[:
-
1
]
...
...
src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py
View file @
b2b3b1a8
...
...
@@ -691,7 +691,6 @@ class LDMBertModel(LDMBertPreTrainedModel):
output_hidden_states
=
None
,
return_dict
=
None
,
):
outputs
=
self
.
model
(
input_ids
,
attention_mask
=
attention_mask
,
...
...
src/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py
View file @
b2b3b1a8
...
...
@@ -38,7 +38,6 @@ class LDMPipeline(DiffusionPipeline):
return_dict
:
bool
=
True
,
**
kwargs
,
)
->
Union
[
Tuple
,
ImagePipelineOutput
]:
r
"""
Args:
batch_size (`int`, *optional*, defaults to 1):
...
...
src/diffusers/schedulers/scheduling_ddpm.py
View file @
b2b3b1a8
...
...
@@ -94,7 +94,6 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
clip_sample
:
bool
=
True
,
tensor_format
:
str
=
"pt"
,
):
if
trained_betas
is
not
None
:
self
.
betas
=
np
.
asarray
(
trained_betas
)
elif
beta_schedule
==
"linear"
:
...
...
@@ -251,7 +250,6 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
noise
:
Union
[
torch
.
FloatTensor
,
np
.
ndarray
],
timesteps
:
Union
[
torch
.
IntTensor
,
np
.
ndarray
],
)
->
Union
[
torch
.
FloatTensor
,
np
.
ndarray
]:
sqrt_alpha_prod
=
self
.
alphas_cumprod
[
timesteps
]
**
0.5
sqrt_alpha_prod
=
self
.
match_shape
(
sqrt_alpha_prod
,
original_samples
)
sqrt_one_minus_alpha_prod
=
(
1
-
self
.
alphas_cumprod
[
timesteps
])
**
0.5
...
...
src/diffusers/schedulers/scheduling_sde_vp.py
View file @
b2b3b1a8
...
...
@@ -40,7 +40,6 @@ class ScoreSdeVpScheduler(SchedulerMixin, ConfigMixin):
@
register_to_config
def
__init__
(
self
,
num_train_timesteps
=
2000
,
beta_min
=
0.1
,
beta_max
=
20
,
sampling_eps
=
1e-3
,
tensor_format
=
"np"
):
self
.
sigmas
=
None
self
.
discrete_sigmas
=
None
self
.
timesteps
=
None
...
...
src/diffusers/utils/logging.py
View file @
b2b3b1a8
...
...
@@ -65,17 +65,14 @@ def _get_default_logging_level():
def
_get_library_name
()
->
str
:
return
__name__
.
split
(
"."
)[
0
]
def
_get_library_root_logger
()
->
logging
.
Logger
:
return
logging
.
getLogger
(
_get_library_name
())
def
_configure_library_root_logger
()
->
None
:
global
_default_handler
with
_lock
:
...
...
@@ -93,7 +90,6 @@ def _configure_library_root_logger() -> None:
def
_reset_library_root_logger
()
->
None
:
global
_default_handler
with
_lock
:
...
...
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