From d075edb6ba07eafd1a54d3d6f8f9de5aaf46febc Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Thu, 10 Oct 2024 06:48:27 -0400
Subject: [PATCH] update

Signed-off-by: Vladimir Mandic <mandic00@live.com>
---
 modules/processing.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/processing.py b/modules/processing.py
index 4f24d4a8e..471053f50 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -50,6 +50,8 @@ class Processed:
         self.image_cfg_scale = p.image_cfg_scale or 0
         self.steps = p.steps or 0
         self.batch_size = max(1, p.batch_size)
+        self.restore_faces = p.restore_faces or False
+        self.face_restoration_model = shared.opts.face_restoration_model if p.restore_faces else None
         self.detailer = p.detailer or False
         self.detailer_model = shared.opts.detailer_model if p.detailer else None
         self.sd_model_hash = getattr(shared.sd_model, 'sd_model_hash', '') if model_data.sd_model is not None else ''