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 17:00:03 +00:00
parent 8aeceb9959
commit a5d4389fdb
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
predict.py
View file

@ -84,10 +84,10 @@ class Predictor(BasePredictor):
k: create_SUPIR_model("options/SUPIR_v0.yaml", SUPIR_sign=k).to( k: create_SUPIR_model("options/SUPIR_v0.yaml", SUPIR_sign=k).to(
self.supir_device self.supir_device
) )
for k in ["Q", "F"][1:] for k in ["Q", "F"]
} }
for k in ["Q", "F"][1:]: for k in ["Q", "F"]:
self.models[k].ae_dtype = convert_dtype(ae_dtype) self.models[k].ae_dtype = convert_dtype(ae_dtype)
self.models[k].model.dtype = convert_dtype(diff_dtype) self.models[k].model.dtype = convert_dtype(diff_dtype)