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