Skip to content

H
Hydra Node Http
Commit 81a39ebb authored by kurumuz's avatar kurumuz
Browse files
Options

fix

parent 724d9a3a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
hydra_node/models.py
View file @ 81a39ebb
...@@ -197,7 +197,7 @@ class StableDiffusionModel(nn.Module): ...@@ -197,7 +197,7 @@ class StableDiffusionModel(nn.Module):
x, y = model.first_stage_model.decoder.load_state_dict(dec_ckpt) x, y = model.first_stage_model.decoder.load_state_dict(dec_ckpt)
model.first_stage_model = model.first_stage_model.float() model.first_stage_model = model.first_stage_model.float()
del ckpt del ckpt
del loss del dec_ckpt
config.logger.info(f"Using VAE from {self.config.vae_path}") config.logger.info(f"Using VAE from {self.config.vae_path}")
if self.config.penultimate == "1": if self.config.penultimate == "1":
......
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