Skip to content

Commit 8178c21

Browse files
authored
Merge pull request #1430 from bghira/bugfix/hidream-vae-scaling
hidream: remove 2x vae scaling
2 parents f243ff0 + 13fe8f2 commit 8178c21

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

helpers/models/hidream/model.py

-4
Original file line numberDiff line numberDiff line change
@@ -111,10 +111,6 @@ def pretrained_load_args(self, pretrained_load_args: dict) -> dict:
111111

112112
return pretrained_load_args
113113

114-
def post_vae_load_setup(self):
115-
# we have to differently scale VAE inputs due to the patches.
116-
self.AUTOENCODER_SCALING_FACTOR *= 2
117-
118114
def _load_pipeline(
119115
self, pipeline_type: str = PipelineTypes.TEXT2IMG, load_base_model: bool = True
120116
):

0 commit comments

Comments
 (0)