wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 21:30:15 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix all undefined variables

Browse Source
This commit is contained in:
Max Tretikov 2024-06-13 14:49:00 -06:00
parent 8be5134f4c
commit c69d4cae0a
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ldm/models/autoencoder.py
View File

@ -1,4 +1,5 @@
import torch import torch
import math
from contextlib import contextmanager from contextlib import contextmanager
from typing import Any, Dict, List, Optional, Tuple, Union from typing import Any, Dict, List, Optional, Tuple, Union
import logging as logpy import logging as logpy
@ -113,7 +114,7 @@ class AutoencodingEngine(AbstractAutoencoder):
self.encoder: torch.nn.Module = instantiate_from_config(encoder_config) self.encoder: torch.nn.Module = instantiate_from_config(encoder_config)
self.decoder: torch.nn.Module = instantiate_from_config(decoder_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 regularizer_config
) )

1
comfy/ldm/modules/diffusionmodules/mmdit.py
View File

@ -1,5 +1,6 @@
import logging import logging
import math import math
from functools import partial
from typing import Dict, Optional from typing import Dict, Optional
import numpy as np import numpy as np