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
e1c72697
Unverified
Commit
e1c72697
authored
Jan 13, 2025
by
Aryan
Committed by
GitHub
Jan 13, 2025
Browse files
Fix Latte output_type (#10558)
update
parent
edb8c1bc
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
1 deletion
+9
-1
src/diffusers/pipelines/latte/pipeline_latte.py
src/diffusers/pipelines/latte/pipeline_latte.py
+9
-1
No files found.
src/diffusers/pipelines/latte/pipeline_latte.py
View file @
e1c72697
...
@@ -30,6 +30,7 @@ from ...schedulers import KarrasDiffusionSchedulers
...
@@ -30,6 +30,7 @@ from ...schedulers import KarrasDiffusionSchedulers
from
...utils
import
(
from
...utils
import
(
BACKENDS_MAPPING
,
BACKENDS_MAPPING
,
BaseOutput
,
BaseOutput
,
deprecate
,
is_bs4_available
,
is_bs4_available
,
is_ftfy_available
,
is_ftfy_available
,
is_torch_xla_available
,
is_torch_xla_available
,
...
@@ -848,7 +849,14 @@ class LattePipeline(DiffusionPipeline):
...
@@ -848,7 +849,14 @@ class LattePipeline(DiffusionPipeline):
if
XLA_AVAILABLE
:
if
XLA_AVAILABLE
:
xm
.
mark_step
()
xm
.
mark_step
()
if
not
output_type
==
"latents"
:
if
output_type
==
"latents"
:
deprecation_message
=
(
"Passing `output_type='latents'` is deprecated. Please pass `output_type='latent'` instead."
)
deprecate
(
"output_type_latents"
,
"1.0.0"
,
deprecation_message
,
standard_warn
=
False
)
output_type
=
"latent"
if
not
output_type
==
"latent"
:
video
=
self
.
decode_latents
(
latents
,
video_length
,
decode_chunk_size
=
14
)
video
=
self
.
decode_latents
(
latents
,
video_length
,
decode_chunk_size
=
14
)
video
=
self
.
video_processor
.
postprocess_video
(
video
=
video
,
output_type
=
output_type
)
video
=
self
.
video_processor
.
postprocess_video
(
video
=
video
,
output_type
=
output_type
)
else
:
else
:
...
...
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