Madiator2011/SUPIR
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

replicate demo

Browse source
This commit is contained in:
chenxwh 2024-02-23 16:57:15 +00:00
parent 8fdcba784a
commit 0d214e30da
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
predict.py
View file

@ -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)