jbilcke-hf HF Staff commited on
Commit
37e49b8
·
1 Parent(s): cdb9a2e
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')