using "none" for no refiner selection
Browse files
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,
|