KevinNg99 commited on
Commit
1a63574
·
1 Parent(s): f129909

disable compile

Browse files
Files changed (1) hide show
  1. hyimage/models/model_zoo.py +3 -3
hyimage/models/model_zoo.py CHANGED
@@ -66,7 +66,7 @@ def HUNYUANIMAGE_V2_1_DIT(**kwargs):
66
  use_cpu_offload=False,
67
  gradient_checkpointing=True,
68
  load_from=f"{HUNYUANIMAGE_V2_1_MODEL_ROOT}/dit/hunyuanimage2.1.safetensors",
69
- use_compile=True,
70
  )
71
 
72
 
@@ -77,7 +77,7 @@ def HUNYUANIMAGE_V2_1_DIT_CFG_DISTILL(**kwargs):
77
  use_cpu_offload=False,
78
  gradient_checkpointing=True,
79
  load_from=f"{HUNYUANIMAGE_V2_1_MODEL_ROOT}/dit/hunyuanimage2.1-distilled.safetensors",
80
- use_compile=True,
81
  )
82
 
83
  # =============================================================================
@@ -91,7 +91,7 @@ def HUNYUANIMAGE_REFINER_DIT(**kwargs):
91
  use_cpu_offload=False,
92
  gradient_checkpointing=True,
93
  load_from=f"{HUNYUANIMAGE_V2_1_MODEL_ROOT}/dit/hunyuanimage-refiner.safetensors",
94
- use_compile=True,
95
  )
96
 
97
  def HUNYUANIMAGE_REFINER_VAE_16x(**kwargs):
 
66
  use_cpu_offload=False,
67
  gradient_checkpointing=True,
68
  load_from=f"{HUNYUANIMAGE_V2_1_MODEL_ROOT}/dit/hunyuanimage2.1.safetensors",
69
+ use_compile=False,
70
  )
71
 
72
 
 
77
  use_cpu_offload=False,
78
  gradient_checkpointing=True,
79
  load_from=f"{HUNYUANIMAGE_V2_1_MODEL_ROOT}/dit/hunyuanimage2.1-distilled.safetensors",
80
+ use_compile=False,
81
  )
82
 
83
  # =============================================================================
 
91
  use_cpu_offload=False,
92
  gradient_checkpointing=True,
93
  load_from=f"{HUNYUANIMAGE_V2_1_MODEL_ROOT}/dit/hunyuanimage-refiner.safetensors",
94
+ use_compile=False,
95
  )
96
 
97
  def HUNYUANIMAGE_REFINER_VAE_16x(**kwargs):