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
91925fbb
Unverified
Commit
91925fbb
authored
Feb 15, 2023
by
Patrick von Platen
Committed by
GitHub
Feb 14, 2023
Browse files
Fix callback type hints - no optional function argument (#2357)
replace type hints
parent
0db19da0
Changes
40
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
22 additions
and
22 deletions
+22
-22
src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py
...table_diffusion/pipeline_onnx_stable_diffusion_inpaint.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint_legacy.py
...iffusion/pipeline_onnx_stable_diffusion_inpaint_legacy.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
...s/pipelines/stable_diffusion/pipeline_stable_diffusion.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py
...s/stable_diffusion/pipeline_stable_diffusion_depth2img.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py
...le_diffusion/pipeline_stable_diffusion_image_variation.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py
...nes/stable_diffusion/pipeline_stable_diffusion_img2img.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py
...nes/stable_diffusion/pipeline_stable_diffusion_inpaint.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
...ble_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py
...e_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py
...stable_diffusion/pipeline_stable_diffusion_k_diffusion.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py
...ble_diffusion/pipeline_stable_diffusion_latent_upscale.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py
...nes/stable_diffusion/pipeline_stable_diffusion_upscale.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py
...sers/pipelines/stable_diffusion/pipeline_stable_unclip.py
+1
-1
src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py
...elines/stable_diffusion/pipeline_stable_unclip_img2img.py
+1
-1
src/diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py
...s/stable_diffusion_safe/pipeline_stable_diffusion_safe.py
+1
-1
src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion.py
...lines/versatile_diffusion/pipeline_versatile_diffusion.py
+3
-3
src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py
...ile_diffusion/pipeline_versatile_diffusion_dual_guided.py
+1
-1
src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py
...diffusion/pipeline_versatile_diffusion_image_variation.py
+1
-1
src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py
...e_diffusion/pipeline_versatile_diffusion_text_to_image.py
+1
-1
src/diffusers/pipelines/vq_diffusion/pipeline_vq_diffusion.py
...diffusers/pipelines/vq_diffusion/pipeline_vq_diffusion.py
+1
-1
No files found.
src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py
View file @
91925fbb
...
...
@@ -257,7 +257,7 @@ class OnnxStableDiffusionInpaintPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
np
.
ndarray
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint_legacy.py
View file @
91925fbb
...
...
@@ -239,7 +239,7 @@ class OnnxStableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
np
.
ndarray
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
**
kwargs
,
):
r
"""
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
View file @
91925fbb
...
...
@@ -472,7 +472,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
cross_attention_kwargs
:
Optional
[
Dict
[
str
,
Any
]]
=
None
,
):
r
"""
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py
View file @
91925fbb
...
...
@@ -497,7 +497,7 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py
View file @
91925fbb
...
...
@@ -274,7 +274,7 @@ class StableDiffusionImageVariationPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py
View file @
91925fbb
...
...
@@ -543,7 +543,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
**
kwargs
,
):
r
"""
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py
View file @
91925fbb
...
...
@@ -609,7 +609,7 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
View file @
91925fbb
...
...
@@ -495,7 +495,7 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
**
kwargs
,
):
r
"""
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py
View file @
91925fbb
...
...
@@ -141,7 +141,7 @@ class StableDiffusionInstructPix2PixPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py
View file @
91925fbb
...
...
@@ -364,7 +364,7 @@ class StableDiffusionKDiffusionPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py
View file @
91925fbb
...
...
@@ -291,7 +291,7 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py
View file @
91925fbb
...
...
@@ -388,7 +388,7 @@ class StableDiffusionUpscalePipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py
View file @
91925fbb
...
...
@@ -620,7 +620,7 @@ class StableUnCLIPPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
cross_attention_kwargs
:
Optional
[
Dict
[
str
,
Any
]]
=
None
,
noise_level
:
int
=
0
,
# prior args
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py
View file @
91925fbb
...
...
@@ -587,7 +587,7 @@ class StableUnCLIPImg2ImgPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
cross_attention_kwargs
:
Optional
[
Dict
[
str
,
Any
]]
=
None
,
noise_level
:
int
=
0
,
image_embeds
:
Optional
[
torch
.
FloatTensor
]
=
None
,
...
...
src/diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py
View file @
91925fbb
...
...
@@ -512,7 +512,7 @@ class StableDiffusionPipelineSafe(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
sld_guidance_scale
:
Optional
[
float
]
=
1000
,
sld_warmup_steps
:
Optional
[
int
]
=
10
,
sld_threshold
:
Optional
[
float
]
=
0.01
,
...
...
src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion.py
View file @
91925fbb
...
...
@@ -95,7 +95,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
@@ -211,7 +211,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
@@ -324,7 +324,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
):
r
"""
Function invoked when calling the pipeline for generation.
...
...
src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py
View file @
91925fbb
...
...
@@ -419,7 +419,7 @@ class VersatileDiffusionDualGuidedPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
**
kwargs
,
):
r
"""
...
...
src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py
View file @
91925fbb
...
...
@@ -271,7 +271,7 @@ class VersatileDiffusionImageVariationPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
**
kwargs
,
):
r
"""
...
...
src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py
View file @
91925fbb
...
...
@@ -353,7 +353,7 @@ class VersatileDiffusionTextToImagePipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
**
kwargs
,
):
r
"""
...
...
src/diffusers/pipelines/vq_diffusion/pipeline_vq_diffusion.py
View file @
91925fbb
...
...
@@ -176,7 +176,7 @@ class VQDiffusionPipeline(DiffusionPipeline):
output_type
:
Optional
[
str
]
=
"pil"
,
return_dict
:
bool
=
True
,
callback
:
Optional
[
Callable
[[
int
,
int
,
torch
.
FloatTensor
],
None
]]
=
None
,
callback_steps
:
Optional
[
int
]
=
1
,
callback_steps
:
int
=
1
,
)
->
Union
[
ImagePipelineOutput
,
Tuple
]:
"""
Function invoked when calling the pipeline for generation.
...
...
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