From 0d214e30dad29c23efbbd9f3664e1d50e90260d2 Mon Sep 17 00:00:00 2001 From: chenxwh Date: Fri, 23 Feb 2024 16:57:15 +0000 Subject: [PATCH] replicate demo --- predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict.py b/predict.py index 51e166f..71442b4 100644 --- a/predict.py +++ b/predict.py @@ -171,7 +171,7 @@ class Predictor(BasePredictor): seed = int.from_bytes(os.urandom(2), "big") print(f"Using seed: {seed}") - model = self.models["Q"] # if model_name == "SUPIR-v0Q" else self.models["F"] + model = self.models["Q"] if model_name == "SUPIR-v0Q" else self.models["F"] lq_img = Image.open(str(image)) lq_img, h0, w0 = PIL2Tensor(lq_img, upsacle=upscale, min_size=min_size)