add the missing get_first_stage_encoding function

2024-06-28 08:10:32 +03:00 · 2024-06-28 08:10:32 +03:00 · 0c7bdcc1b1
parent fc8b126673
commit 0c7bdcc1b1
1 changed files with 3 additions and 0 deletions
--- 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)