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

using "none" for no refiner selection

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -147,7 +147,7 @@ def run_inference(config, config_history, progress=gr.Progress(track_tqdm=True))
147
  torch_dtype = get_data_type(config["data_type"]),
148
  variant = get_variant(config["variant"])).to(config["device"])
149
 
150
- if config['refiner'] != '':
151
  refiner = DiffusionPipeline.from_pretrained(
152
  config['refiner'],
153
  text_encoder_2=pipeline.text_encoder_2,
 
147
  torch_dtype = get_data_type(config["data_type"]),
148
  variant = get_variant(config["variant"])).to(config["device"])
149
 
150
+ if config['refiner'].lower() != 'none':
151
  refiner = DiffusionPipeline.from_pretrained(
152
  config['refiner'],
153
  text_encoder_2=pipeline.text_encoder_2,