Spaces:
Running
on
A100
Running
on
A100
disable compile
Browse files
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=
|
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=
|
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=
|
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):
|