From 13bd2d3d29c66da24fac55ef82495a099c471884 Mon Sep 17 00:00:00 2001 From: pnsuau Date: Mon, 3 Jul 2023 10:05:52 +0000 Subject: [PATCH] fix: inference on ref cond --- models/palette_model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/models/palette_model.py b/models/palette_model.py index fc49e177d..bce3804f4 100644 --- a/models/palette_model.py +++ b/models/palette_model.py @@ -271,7 +271,10 @@ def __init__(self, opt, rank): for i in range(self.nb_classes_inference): self.gen_visual_names.append("output_" + str(i + 1) + "_") - elif self.opt.alg_palette_cond_image_creation == "ref": + elif ( + self.opt.alg_palette_cond_image_creation == "ref" + or "ref" in self.opt.alg_palette_conditioning + ): for i in range(self.inference_num): self.gen_visual_names.append("cond_ref_" + str(i + 1) + "_") self.gen_visual_names.append("output_" + str(i + 1) + "_")