← back to Exo

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

17 lines

"""
This type stub file was generated by pyright.
"""

import mlx.core as mx
from typing import Dict, List, Optional
from mflux.models.common.weights.mapping.weight_mapping import WeightTarget

class WeightMapper:
    @staticmethod
    def apply_mapping(
        hf_weights: Dict[str, mx.array],
        mapping: List[WeightTarget],
        num_blocks: Optional[int] = ...,
        num_layers: Optional[int] = ...,
    ) -> Dict: ...