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
b3911f89
Commit
b3911f89
authored
Nov 17, 2022
by
Patrick von Platen
Browse files
make fix copies
parent
245e9cc7
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
6 additions
and
6 deletions
+6
-6
src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py
...ffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py
+2
-2
src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py
...rs/pipelines/stable_diffusion/pipeline_cycle_diffusion.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
No files found.
src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py
View file @
b3911f89
...
@@ -179,7 +179,7 @@ class AltDiffusionPipeline(DiffusionPipeline):
...
@@ -179,7 +179,7 @@ class AltDiffusionPipeline(DiffusionPipeline):
# set slice_size = `None` to disable `attention slicing`
# set slice_size = `None` to disable `attention slicing`
self
.
enable_attention_slicing
(
None
)
self
.
enable_attention_slicing
(
None
)
def
enable_sequential_cpu_offload
(
self
):
def
enable_sequential_cpu_offload
(
self
,
gpu_id
=
0
):
r
"""
r
"""
Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet,
Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet,
text_encoder, vae and safety checker have their state dicts saved to CPU and then are moved to a
text_encoder, vae and safety checker have their state dicts saved to CPU and then are moved to a
...
@@ -190,7 +190,7 @@ class AltDiffusionPipeline(DiffusionPipeline):
...
@@ -190,7 +190,7 @@ class AltDiffusionPipeline(DiffusionPipeline):
else
:
else
:
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
device
=
torch
.
device
(
"cuda"
)
device
=
torch
.
device
(
f
"cuda
:
{
gpu_id
}
"
)
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
if
cpu_offloaded_model
is
not
None
:
if
cpu_offloaded_model
is
not
None
:
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py
View file @
b3911f89
...
@@ -220,7 +220,7 @@ class CycleDiffusionPipeline(DiffusionPipeline):
...
@@ -220,7 +220,7 @@ class CycleDiffusionPipeline(DiffusionPipeline):
else
:
else
:
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
device
=
torch
.
device
(
"cuda"
)
device
=
torch
.
device
(
f
"cuda
:
{
gpu_id
}
"
)
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
if
cpu_offloaded_model
is
not
None
:
if
cpu_offloaded_model
is
not
None
:
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py
View file @
b3911f89
...
@@ -187,7 +187,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline):
...
@@ -187,7 +187,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline):
else
:
else
:
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
device
=
torch
.
device
(
"cuda"
)
device
=
torch
.
device
(
f
"cuda
:
{
gpu_id
}
"
)
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
if
cpu_offloaded_model
is
not
None
:
if
cpu_offloaded_model
is
not
None
:
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py
View file @
b3911f89
...
@@ -180,7 +180,7 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline):
...
@@ -180,7 +180,7 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline):
else
:
else
:
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
device
=
torch
.
device
(
"cuda"
)
device
=
torch
.
device
(
f
"cuda
:
{
gpu_id
}
"
)
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
if
cpu_offloaded_model
is
not
None
:
if
cpu_offloaded_model
is
not
None
:
...
...
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
View file @
b3911f89
...
@@ -200,7 +200,7 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
...
@@ -200,7 +200,7 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
else
:
else
:
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
raise
ImportError
(
"Please install accelerate via `pip install accelerate`"
)
device
=
torch
.
device
(
"cuda"
)
device
=
torch
.
device
(
f
"cuda
:
{
gpu_id
}
"
)
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
for
cpu_offloaded_model
in
[
self
.
unet
,
self
.
text_encoder
,
self
.
vae
,
self
.
safety_checker
]:
if
cpu_offloaded_model
is
not
None
:
if
cpu_offloaded_model
is
not
None
:
...
...
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