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
fc94c60c
Unverified
Commit
fc94c60c
authored
Dec 10, 2022
by
Patrick von Platen
Committed by
GitHub
Dec 10, 2022
Browse files
Remove unnecessary offset in img2img (#1653)
remove unnecessary offset in img2img
parent
ea64a786
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
10 additions
and
20 deletions
+10
-20
src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py
...pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py
+2
-4
src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py
...rs/pipelines/stable_diffusion/pipeline_cycle_diffusion.py
+2
-4
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py
...s/stable_diffusion/pipeline_stable_diffusion_depth2img.py
+2
-4
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py
...nes/stable_diffusion/pipeline_stable_diffusion_img2img.py
+2
-4
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
...ble_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
+2
-4
No files found.
src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py
View file @
fc94c60c
...
@@ -376,11 +376,9 @@ class AltDiffusionImg2ImgPipeline(DiffusionPipeline):
...
@@ -376,11 +376,9 @@ class AltDiffusionImg2ImgPipeline(DiffusionPipeline):
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
# get the original timestep using init_timestep
# get the original timestep using init_timestep
offset
=
self
.
scheduler
.
config
.
get
(
"steps_offset"
,
0
)
init_timestep
=
min
(
int
(
num_inference_steps
*
strength
),
num_inference_steps
)
init_timestep
=
int
(
num_inference_steps
*
strength
)
+
offset
init_timestep
=
min
(
init_timestep
,
num_inference_steps
)
t_start
=
max
(
num_inference_steps
-
init_timestep
+
offset
,
0
)
t_start
=
max
(
num_inference_steps
-
init_timestep
,
0
)
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
return
timesteps
,
num_inference_steps
-
t_start
return
timesteps
,
num_inference_steps
-
t_start
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py
View file @
fc94c60c
...
@@ -414,11 +414,9 @@ class CycleDiffusionPipeline(DiffusionPipeline):
...
@@ -414,11 +414,9 @@ class CycleDiffusionPipeline(DiffusionPipeline):
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.StableDiffusionImg2ImgPipeline.get_timesteps
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.StableDiffusionImg2ImgPipeline.get_timesteps
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
# get the original timestep using init_timestep
# get the original timestep using init_timestep
offset
=
self
.
scheduler
.
config
.
get
(
"steps_offset"
,
0
)
init_timestep
=
min
(
int
(
num_inference_steps
*
strength
),
num_inference_steps
)
init_timestep
=
int
(
num_inference_steps
*
strength
)
+
offset
init_timestep
=
min
(
init_timestep
,
num_inference_steps
)
t_start
=
max
(
num_inference_steps
-
init_timestep
+
offset
,
0
)
t_start
=
max
(
num_inference_steps
-
init_timestep
,
0
)
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
return
timesteps
,
num_inference_steps
-
t_start
return
timesteps
,
num_inference_steps
-
t_start
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py
View file @
fc94c60c
...
@@ -323,11 +323,9 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline):
...
@@ -323,11 +323,9 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline):
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.StableDiffusionImg2ImgPipeline.get_timesteps
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.StableDiffusionImg2ImgPipeline.get_timesteps
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
# get the original timestep using init_timestep
# get the original timestep using init_timestep
offset
=
self
.
scheduler
.
config
.
get
(
"steps_offset"
,
0
)
init_timestep
=
min
(
int
(
num_inference_steps
*
strength
),
num_inference_steps
)
init_timestep
=
int
(
num_inference_steps
*
strength
)
+
offset
init_timestep
=
min
(
init_timestep
,
num_inference_steps
)
t_start
=
max
(
num_inference_steps
-
init_timestep
+
offset
,
0
)
t_start
=
max
(
num_inference_steps
-
init_timestep
,
0
)
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
return
timesteps
,
num_inference_steps
-
t_start
return
timesteps
,
num_inference_steps
-
t_start
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py
View file @
fc94c60c
...
@@ -381,11 +381,9 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline):
...
@@ -381,11 +381,9 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline):
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
# get the original timestep using init_timestep
# get the original timestep using init_timestep
offset
=
self
.
scheduler
.
config
.
get
(
"steps_offset"
,
0
)
init_timestep
=
min
(
int
(
num_inference_steps
*
strength
),
num_inference_steps
)
init_timestep
=
int
(
num_inference_steps
*
strength
)
+
offset
init_timestep
=
min
(
init_timestep
,
num_inference_steps
)
t_start
=
max
(
num_inference_steps
-
init_timestep
+
offset
,
0
)
t_start
=
max
(
num_inference_steps
-
init_timestep
,
0
)
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
return
timesteps
,
num_inference_steps
-
t_start
return
timesteps
,
num_inference_steps
-
t_start
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
View file @
fc94c60c
...
@@ -396,11 +396,9 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
...
@@ -396,11 +396,9 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.StableDiffusionImg2ImgPipeline.get_timesteps
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.StableDiffusionImg2ImgPipeline.get_timesteps
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
def
get_timesteps
(
self
,
num_inference_steps
,
strength
,
device
):
# get the original timestep using init_timestep
# get the original timestep using init_timestep
offset
=
self
.
scheduler
.
config
.
get
(
"steps_offset"
,
0
)
init_timestep
=
min
(
int
(
num_inference_steps
*
strength
),
num_inference_steps
)
init_timestep
=
int
(
num_inference_steps
*
strength
)
+
offset
init_timestep
=
min
(
init_timestep
,
num_inference_steps
)
t_start
=
max
(
num_inference_steps
-
init_timestep
+
offset
,
0
)
t_start
=
max
(
num_inference_steps
-
init_timestep
,
0
)
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
timesteps
=
self
.
scheduler
.
timesteps
[
t_start
:]
return
timesteps
,
num_inference_steps
-
t_start
return
timesteps
,
num_inference_steps
-
t_start
...
...
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