n42 commited on
Commit
b5eaa29
·
1 Parent(s): ffd045a

using "none" for no refiner selection

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -144,7 +144,7 @@ def assemble_code(str_config):
144
 
145
  if str(config["use_safetensors"]).lower() == 'false': code['051_cpu_offload'] = "pipeline.enable_model_cpu_offload()"
146
 
147
- if config['refiner'] != '':
148
  code['052_refiner'] = f'''refiner = DiffusionPipeline.from_pretrained(
149
  "{config['refiner']}",
150
  text_encoder_2 = base.text_encoder_2,
 
144
 
145
  if str(config["use_safetensors"]).lower() == 'false': code['051_cpu_offload'] = "pipeline.enable_model_cpu_offload()"
146
 
147
+ if config['refiner'].lower() != 'none':
148
  code['052_refiner'] = f'''refiner = DiffusionPipeline.from_pretrained(
149
  "{config['refiner']}",
150
  text_encoder_2 = base.text_encoder_2,