Update low memory table

2024-12-24 16:35:14 +08:00
parent 405a4ddc3a
commit 5c690e1ed9
5 changed files with 239 additions and 8 deletions
@@ -201,10 +201,7 @@ else:
    height, width = video_size

 # Set hidden states offload steps
-if pipeline.model_cpu_offload_flag:
-    pipeline.transformer.hidden_cache_size = gpu_offload_steps
-else:
-    pipeline.transformer.hidden_cache_size = 0
+pipeline.transformer.hidden_cache_size = gpu_offload_steps

 # Load Sampler
 if scheduler_name == "DPM++":