nickyreinert-vml commited on
Commit
a0e2e28
1 Parent(s): e4a20eb

fix wrong quotes

Browse files
Files changed (1) hide show
  1. config.py +1 -1
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="reduce-overhead", fullgraph=True)")
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