hpoghos commited on
Commit
4358fb0
1 Parent(s): 566e79d

Update streaming_svd_inference.py

Browse files
Files changed (1) hide show
  1. streaming_svd_inference.py +1 -1
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()