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 |
|