Commit 0c7bdcc1 authored by AUTOMATIC1111's avatar AUTOMATIC1111

add the missing get_first_stage_encoding function

parent fc8b1266
......@@ -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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment