replicate demo

chenxwh · Feb 23, 2024 · a5d4389 · a5d4389
1 parent 8aeceb9
commit a5d4389
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/predict.py b/predict.py
@@ -84,10 +84,10 @@ def setup(self) -> None:
             k: create_SUPIR_model("options/SUPIR_v0.yaml", SUPIR_sign=k).to(
                 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].model.dtype = convert_dtype(diff_dtype)