OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-12-29 19:05:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

add the missing get_first_stage_encoding function

Browse Source
This commit is contained in:
AUTOMATIC1111 2024-06-28 08:10:32 +03:00
parent fc8b126673
commit 0c7bdcc1b1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/models/sd3/sd3_model.py
View File

@ -61,6 +61,9 @@ class SD3Inferencer(torch.nn.Module):
latent = self.first_stage_model.encode(image) latent = self.first_stage_model.encode(image)
return self.latent_format.process_in(latent) return self.latent_format.process_in(latent)
def get_first_stage_encoding(self, x):
return x
def create_denoiser(self): def create_denoiser(self):
return SD3Denoiser(self, self.model.model_sampling.sigmas) return SD3Denoiser(self, self.model.model_sampling.sigmas)