nickyreinert-vml commited on
Commit ·
a0e2e28
1
Parent(s): e4a20eb
fix wrong quotes
Browse files
config.py
CHANGED
|
@@ -177,7 +177,7 @@ def assemble_code(str_config):
|
|
| 177 |
variant=variant).to(device)''')
|
| 178 |
|
| 179 |
if str(config["attention_slicing"]).lower() != 'false': code.append("pipeline.enable_attention_slicing()")
|
| 180 |
-
if str(config["pre_compile_unet"]).lower() != 'false': code.append("pipeline.unet = torch.compile(pipeline.unet, mode=
|
| 181 |
|
| 182 |
if str(config["cpu_offload"]).lower() != 'false': code.append("pipeline.enable_model_cpu_offload()")
|
| 183 |
|
|
|
|
| 177 |
variant=variant).to(device)''')
|
| 178 |
|
| 179 |
if str(config["attention_slicing"]).lower() != 'false': code.append("pipeline.enable_attention_slicing()")
|
| 180 |
+
if str(config["pre_compile_unet"]).lower() != 'false': code.append("pipeline.unet = torch.compile(pipeline.unet, mode='reduce-overhead', fullgraph=True)")
|
| 181 |
|
| 182 |
if str(config["cpu_offload"]).lower() != 'false': code.append("pipeline.enable_model_cpu_offload()")
|
| 183 |
|