Spaces:
Running
on
A100
Running
on
A100
Commit
·
37e49b8
1
Parent(s):
cdb9a2e
test
Browse files
hymm_sp/diffusion/pipelines/pipeline_hunyuan_video_game.py
CHANGED
@@ -989,7 +989,7 @@ class HunyuanVideoGamePipeline(DiffusionPipeline):
|
|
989 |
is_cache = False
|
990 |
else:
|
991 |
is_cache = True
|
992 |
-
if latent_model_input.shape[-1]*latent_model_input.shape[-2]>64*112 and cpu_offload:
|
993 |
if i==0:
|
994 |
print(f'cpu_offload={cpu_offload} and \
|
995 |
{latent_model_input.shape[-2:]} is large, split infer noise-pred')
|
|
|
989 |
is_cache = False
|
990 |
else:
|
991 |
is_cache = True
|
992 |
+
if latent_model_input.shape[-1]*latent_model_input.shape[-2]>64*112 and cpu_offload and self.do_classifier_free_guidance:
|
993 |
if i==0:
|
994 |
print(f'cpu_offload={cpu_offload} and \
|
995 |
{latent_model_input.shape[-2:]} is large, split infer noise-pred')
|