Remove vae type alias from runtime_config.py
This commit is contained in:
parent
429b580b04
commit
f08c667e1a
@ -1,5 +1,4 @@
|
|||||||
import logging
|
import logging
|
||||||
import typing as t
|
|
||||||
|
|
||||||
import mlx.core as mx
|
import mlx.core as mx
|
||||||
import numpy as np
|
import numpy as np
|
||||||
@ -9,8 +8,6 @@ from mflux.config.model_config import ModelConfig
|
|||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
VAE: t.TypeAlias = "mflux.models.vae.vae.VAE" # noqa: F821
|
|
||||||
|
|
||||||
|
|
||||||
class RuntimeConfig:
|
class RuntimeConfig:
|
||||||
def __init__(self, config: Config | ConfigControlnet, model_config):
|
def __init__(self, config: Config | ConfigControlnet, model_config):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user