Julian Bilcke
commited on
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')
|