From 0c7bdcc1b130130bdaec425a9c3537ba098e0098 Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Fri, 28 Jun 2024 08:10:32 +0300
Subject: [PATCH] add the missing get_first_stage_encoding function

---
 modules/models/sd3/sd3_model.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/models/sd3/sd3_model.py b/modules/models/sd3/sd3_model.py
index 10209f82a..f8a4c96cc 100644
--- a/modules/models/sd3/sd3_model.py
+++ b/modules/models/sd3/sd3_model.py
@@ -61,6 +61,9 @@ class SD3Inferencer(torch.nn.Module):
         latent = self.first_stage_model.encode(image)
         return self.latent_format.process_in(latent)
 
+    def get_first_stage_encoding(self, x):
+        return x
+
     def create_denoiser(self):
         return SD3Denoiser(self, self.model.model_sampling.sigmas)