Spaces:
Running
on
Zero
Running
on
Zero
Update streaming_svd_inference.py
Browse files
streaming_svd_inference.py
CHANGED
@@ -87,7 +87,7 @@ class StreamingSVD():
|
|
87 |
|
88 |
enhance_pipeline, enhance_generator = i2v_enhance_interface.i2v_enhance_init(
|
89 |
model.i2v_enhance)
|
90 |
-
|
91 |
flux_pipe = FluxPipeline.from_pretrained(
|
92 |
"black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
|
93 |
flux_pipe.enable_model_cpu_offload()
|
|
|
87 |
|
88 |
enhance_pipeline, enhance_generator = i2v_enhance_interface.i2v_enhance_init(
|
89 |
model.i2v_enhance)
|
90 |
+
enhance_pipeline.unet.enable_forward_chunking(chunk_size=1, dim=1)
|
91 |
flux_pipe = FluxPipeline.from_pretrained(
|
92 |
"black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
|
93 |
flux_pipe.enable_model_cpu_offload()
|