add the missing get_first_stage_encoding function

This commit is contained in:
AUTOMATIC1111 2024-06-28 08:10:32 +03:00
parent fc8b126673
commit 0c7bdcc1b1

View File

@ -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)