feat: support codeformer face restoration model (#23)

stable_horde.py

@@ -161,7 +161,7 @@ async def get_popped_request(self) -> Optional[Dict[str, Any]]:
             "nsfw": self.config.nsfw,
             "blacklist": [],
             "models": self.config.models,
-            "bridge_version": 8,
+            "bridge_version": 9,
             "threads": 1,
             "max_pixels": self.config.max_pixels,
             "allow_img2img": self.config.allow_img2img,
@@ -207,6 +207,15 @@ async def handle_request(self, req: Dict[str, Any]):
             negative = ""
 
         postprocessors = req['payload'].get('post_processing', None) or []
+        restore_faces = False
+        if "GFPGAN" in postprocessors:
+            restore_faces = True
+            shared.opts.face_restoration_model = "GFPGAN"
+            shared.opts.save(shared.config_filename)
+        elif "CodeFormers" in postprocessors:
+            restore_faces = True
+            shared.opts.face_restoration_model = "CodeFormer"
+            shared.opts.save(shared.config_filename)
 
         params = {
             "sd_model": shared.sd_model,
@@ -221,7 +230,7 @@ async def handle_request(self, req: Dict[str, Any]):
             "subseed": req['payload'].get('seed_variation', 1),
             "steps": req['payload']['ddim_steps'],
             "n_iter": req['payload']['n_iter'],
-            "restore_faces": "GFPGAN" in postprocessors,
+            "restore_faces": restore_faces,
             "do_not_save_samples": True,
             "do_not_save_grid": True,
         }