Remove vae type alias from runtime_config.py

2024-10-20 11:05:45 +02:00 · 2024-10-20 11:05:45 +02:00 · f08c667e1a
commit f08c667e1a
parent 429b580b04
1 changed files with 0 additions and 3 deletions
--- a/src/mflux/config/runtime_config.py
+++ b/src/mflux/config/runtime_config.py
@ -1,5 +1,4 @@
 import logging
-import typing as t

 import mlx.core as mx
 import numpy as np
@ -9,8 +8,6 @@ from mflux.config.model_config import ModelConfig

 logger = logging.getLogger(__name__)

-VAE: t.TypeAlias = "mflux.models.vae.vae.VAE"  # noqa: F821
-

 class RuntimeConfig:
    def __init__(self, config: Config | ConfigControlnet, model_config):