vllm loads!

.mlx_typings/mlx/nn/layers/__init__.pyi

@@ -1,20 +0,0 @@
-"""
-This type stub file was generated by pyright.
-"""
-
-from activations import *
-from base import *
-from containers import *
-from convolution import *
-from convolution_transpose import *
-from distributed import *
-from dropout import *
-from embedding import *
-from linear import *
-from normalization import *
-from pooling import *
-from positional_encoding import *
-from quantized import *
-from recurrent import *
-from transformer import *
-from upsample import *

.typings/mflux/callbacks/callback.pyi

@@ -2,10 +2,11 @@
 This type stub file was generated by pyright.
 """
 
+from typing import Protocol
+
 import mlx.core as mx
 import PIL.Image
 import tqdm
-from typing import Protocol
 from mflux.models.common.config.config import Config
 
 class BeforeLoopCallback(Protocol):

.typings/mflux/callbacks/callback_registry.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import TYPE_CHECKING
+
 from mflux.callbacks.callback import (
     AfterLoopCallback,
     BeforeLoopCallback,

.typings/mflux/callbacks/generation_context.pyi

@@ -2,10 +2,11 @@
 This type stub file was generated by pyright.
 """
 
+from typing import TYPE_CHECKING
+
 import mlx.core as mx
 import PIL.Image
 import tqdm
-from typing import TYPE_CHECKING
 from mflux.callbacks.callback_registry import CallbackRegistry
 from mflux.models.common.config.config import Config
 

.typings/mflux/models/common/config/config.pyi

@@ -2,11 +2,12 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from pathlib import Path
 from typing import Any
-from tqdm import tqdm
+
+import mlx.core as mx
 from mflux.models.common.config.model_config import ModelConfig
+from tqdm import tqdm
 
 logger = ...
 

.typings/mflux/models/common/config/model_config.pyi

@@ -2,10 +2,11 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from functools import lru_cache
 from typing import Literal
 
+import mlx.core as mx
+
 class ModelConfig:
     precision: mx.Dtype = ...
     def __init__(

.typings/mflux/models/common/latent_creator/latent_creator.pyi

@@ -2,10 +2,10 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from pathlib import Path
 from typing import TYPE_CHECKING, TypeAlias
-from mlx import nn
+
+import mlx.core as mx
 from mflux.models.common.vae.tiling_config import TilingConfig
 from mflux.models.fibo.latent_creator.fibo_latent_creator import FiboLatentCreator
 from mflux.models.flux.latent_creator.flux_latent_creator import FluxLatentCreator
@@ -13,6 +13,7 @@ from mflux.models.qwen.latent_creator.qwen_latent_creator import QwenLatentCreat
 from mflux.models.z_image.latent_creator.z_image_latent_creator import (
     ZImageLatentCreator,
 )
+from mlx import nn
 
 if TYPE_CHECKING:
     LatentCreatorType: TypeAlias = type[

.typings/mflux/models/common/lora/layer/fused_linear_lora_layer.pyi

@@ -2,8 +2,8 @@
 This type stub file was generated by pyright.
 """
 
-from mlx import nn
 from mflux.models.common.lora.layer.linear_lora_layer import LoRALinear
+from mlx import nn
 
 class FusedLoRALinear(nn.Module):
     def __init__(

.typings/mflux/models/common/lora/mapping/lora_loader.pyi

@@ -2,10 +2,11 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
-import mlx.nn as nn
 from collections.abc import Callable
 from dataclasses import dataclass
+
+import mlx.core as mx
+import mlx.nn as nn
 from mflux.models.common.lora.mapping.lora_mapping import LoRATarget
 
 @dataclass

.typings/mflux/models/common/lora/mapping/lora_mapping.pyi

@@ -2,11 +2,12 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from collections.abc import Callable
 from dataclasses import dataclass
 from typing import List, Protocol
 
+import mlx.core as mx
+
 @dataclass
 class LoRATarget:
     model_path: str

.typings/mflux/models/common/resolution/actions.pyi

@@ -36,4 +36,3 @@ class Rule(NamedTuple):
     name: str
     check: str
     action: QuantizationAction | PathAction | LoraAction | ConfigAction
-    ...

.typings/mflux/models/common/resolution/config_resolution.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import TYPE_CHECKING
+
 from mflux.models.common.config.model_config import ModelConfig
 
 if TYPE_CHECKING: ...

.typings/mflux/models/common/schedulers/base_scheduler.pyi

@@ -2,9 +2,10 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from abc import ABC, abstractmethod
 
+import mlx.core as mx
+
 class BaseScheduler(ABC):
     @property
     @abstractmethod

.typings/mflux/models/common/schedulers/flow_match_euler_discrete_scheduler.pyi

@@ -2,8 +2,9 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from typing import TYPE_CHECKING
+
+import mlx.core as mx
 from mflux.models.common.config.config import Config
 from mflux.models.common.schedulers.base_scheduler import BaseScheduler
 

.typings/mflux/models/common/schedulers/linear_scheduler.pyi

@@ -2,8 +2,9 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from typing import TYPE_CHECKING
+
+import mlx.core as mx
 from mflux.models.common.config.config import Config
 from mflux.models.common.schedulers.base_scheduler import BaseScheduler
 

.typings/mflux/models/common/schedulers/seedvr2_euler_scheduler.pyi

@@ -2,8 +2,9 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from typing import TYPE_CHECKING
+
+import mlx.core as mx
 from mflux.models.common.config.config import Config
 from mflux.models.common.schedulers.base_scheduler import BaseScheduler
 

.typings/mflux/models/common/tokenizer/tokenizer.pyi

@@ -4,9 +4,10 @@ This type stub file was generated by pyright.
 
 from abc import ABC, abstractmethod
 from typing import Protocol, runtime_checkable
+
+from mflux.models.common.tokenizer.tokenizer_output import TokenizerOutput
 from PIL import Image
 from transformers import PreTrainedTokenizer
-from mflux.models.common.tokenizer.tokenizer_output import TokenizerOutput
 
 """
 This type stub file was generated by pyright.

.typings/mflux/models/common/tokenizer/tokenizer_loader.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import TYPE_CHECKING
+
 from mflux.models.common.tokenizer.tokenizer import BaseTokenizer
 from mflux.models.common.weights.loading.weight_definition import TokenizerDefinition
 

.typings/mflux/models/common/tokenizer/tokenizer_output.pyi

@@ -2,9 +2,10 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from dataclasses import dataclass
 
+import mlx.core as mx
+
 """
 This type stub file was generated by pyright.
 """

.typings/mflux/models/common/vae/vae_tiler.pyi

@@ -2,9 +2,10 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from typing import Callable
 
+import mlx.core as mx
+
 class VAETiler:
     @staticmethod
     def encode_image_tiled(

.typings/mflux/models/common/vae/vae_util.pyi

@@ -3,8 +3,8 @@ This type stub file was generated by pyright.
 """
 
 import mlx.core as mx
-from mlx import nn
 from mflux.models.common.vae.tiling_config import TilingConfig
+from mlx import nn
 
 class VAEUtil:
     @staticmethod

.typings/mflux/models/common/weights/loading/weight_applier.pyi

@@ -2,8 +2,9 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.nn as nn
 from typing import TYPE_CHECKING
+
+import mlx.nn as nn
 from mflux.models.common.weights.loading.loaded_weights import LoadedWeights
 from mflux.models.common.weights.loading.weight_definition import (
     ComponentDefinition,

.typings/mflux/models/common/weights/loading/weight_definition.pyi

@@ -2,11 +2,12 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from dataclasses import dataclass
-from typing import Callable, List, TYPE_CHECKING, TypeAlias
-from mflux.models.common.weights.mapping.weight_mapping import WeightTarget
+from typing import TYPE_CHECKING, Callable, List, TypeAlias
+
+import mlx.core as mx
 from mflux.models.common.tokenizer.tokenizer import BaseTokenizer
+from mflux.models.common.weights.mapping.weight_mapping import WeightTarget
 from mflux.models.depth_pro.weights.depth_pro_weight_definition import (
     DepthProWeightDefinition,
 )

.typings/mflux/models/common/weights/loading/weight_loader.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import TYPE_CHECKING
+
 from mflux.models.common.weights.loading.loaded_weights import LoadedWeights
 from mflux.models.common.weights.loading.weight_definition import (
     ComponentDefinition,

.typings/mflux/models/common/weights/mapping/weight_mapper.pyi

@@ -2,8 +2,9 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from typing import Dict, List, Optional
+
+import mlx.core as mx
 from mflux.models.common.weights.mapping.weight_mapping import WeightTarget
 
 class WeightMapper:

.typings/mflux/models/common/weights/mapping/weight_mapping.pyi

@@ -2,10 +2,11 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from dataclasses import dataclass
 from typing import Callable, List, Optional, Protocol
 
+import mlx.core as mx
+
 """
 This type stub file was generated by pyright.
 """

.typings/mflux/models/common/weights/saving/model_saver.pyi

@@ -2,7 +2,8 @@
 This type stub file was generated by pyright.
 """
 
-from typing import Any, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
+
 from mflux.models.common.weights.loading.weight_definition import WeightDefinitionType
 
 if TYPE_CHECKING: ...

.typings/mflux/models/depth_pro/model/depth_pro.pyi

@@ -2,9 +2,10 @@
 This type stub file was generated by pyright.
 """
 
-import mlx.core as mx
 from dataclasses import dataclass
 from pathlib import Path
+
+import mlx.core as mx
 from PIL import Image
 
 @dataclass
@@ -13,7 +14,6 @@ class DepthResult:
     depth_array: mx.array
     min_depth: float
     max_depth: float
-    ...
 
 class DepthPro:
     def __init__(self, quantize: int | None = ...) -> None: ...

.typings/mflux/models/depth_pro/weights/depth_pro_weight_definition.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import List
+
 from mflux.models.common.weights.loading.weight_definition import (
     ComponentDefinition,
     TokenizerDefinition,

.typings/mflux/models/depth_pro/weights/depth_pro_weight_mapping.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import List
+
 from mflux.models.common.weights.mapping.weight_mapping import (
     WeightMapping,
     WeightTarget,

.typings/mflux/models/fibo/weights/fibo_weight_definition.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import List
+
 from mflux.models.common.weights.loading.weight_definition import (
     ComponentDefinition,
     TokenizerDefinition,

.typings/mflux/models/fibo/weights/fibo_weight_mapping.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import List
+
 from mflux.models.common.weights.mapping.weight_mapping import (
     WeightMapping,
     WeightTarget,

.typings/mflux/models/fibo_vlm/tokenizer/qwen2vl_processor.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import Optional, Union
+
 from PIL import Image
 
 class Qwen2VLProcessor:

.typings/mflux/models/fibo_vlm/weights/fibo_vlm_weight_definition.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import List
+
 from mflux.models.common.weights.loading.weight_definition import (
     ComponentDefinition,
     TokenizerDefinition,

.typings/mflux/models/fibo_vlm/weights/fibo_vlm_weight_mapping.pyi

@@ -3,6 +3,7 @@ This type stub file was generated by pyright.
 """
 
 from typing import List
+
 from mflux.models.common.weights.mapping.weight_mapping import (
     WeightMapping,
     WeightTarget,