Skip to content

GitLab

"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "d32bce08c6d205d0468625c5b308b895b166ac1b"
Unverified Commit 42077e6c authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Fix failing cpu offload test for LTX Latent Upscale (#11755) 

update
parent 3d8d8485
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/pipelines/ltx/pipeline_ltx_latent_upsample.py
View file @ 42077e6c
...@@ -44,6 +44,8 @@ def retrieve_latents( ...@@ -44,6 +44,8 @@ def retrieve_latents(
class LTXLatentUpsamplePipeline(DiffusionPipeline): class LTXLatentUpsamplePipeline(DiffusionPipeline):
model_cpu_offload_seq = ""
def __init__( def __init__(
self, self,
vae: AutoencoderKLLTXVideo, vae: AutoencoderKLLTXVideo,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment