"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
...@@ -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