From c69d4cae0aca263ae9bd48572c51614f7ce4841d Mon Sep 17 00:00:00 2001 From: Max Tretikov Date: Thu, 13 Jun 2024 14:49:00 -0600 Subject: [PATCH] Fix all undefined variables --- comfy/ldm/models/autoencoder.py | 3 ++- comfy/ldm/modules/diffusionmodules/mmdit.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/comfy/ldm/models/autoencoder.py b/comfy/ldm/models/autoencoder.py index a84964410..2f4b02f2e 100644 --- a/comfy/ldm/models/autoencoder.py +++ b/comfy/ldm/models/autoencoder.py @@ -1,4 +1,5 @@ import torch +import math from contextlib import contextmanager from typing import Any, Dict, List, Optional, Tuple, Union import logging as logpy @@ -113,7 +114,7 @@ class AutoencodingEngine(AbstractAutoencoder): self.encoder: torch.nn.Module = instantiate_from_config(encoder_config) self.decoder: torch.nn.Module = instantiate_from_config(decoder_config) - self.regularization: AbstractRegularizer = instantiate_from_config( + self.regularization: DiagonalGaussianRegularizer = instantiate_from_config( regularizer_config ) diff --git a/comfy/ldm/modules/diffusionmodules/mmdit.py b/comfy/ldm/modules/diffusionmodules/mmdit.py index 0cb6bd312..7c49f2946 100644 --- a/comfy/ldm/modules/diffusionmodules/mmdit.py +++ b/comfy/ldm/modules/diffusionmodules/mmdit.py @@ -1,5 +1,6 @@ import logging import math +from functools import partial from typing import Dict, Optional import numpy as np