Spaces:
Running
Running
File size: 842 Bytes
56dc4d0 0f62aee 3397d41 0f62aee 5b56303 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
---
title: README
emoji: ⚡
colorFrom: blue
colorTo: green
sdk: static
pinned: false
---
# **MLX Vision**
A community org for model weights compatible with (mlx-im)[https://github.com/riccardomusmeci/mlx-im/tree/main] powered by MLX.
These are weights converted from timm/torchvision and ready to be used.
## **How to install**
```
git clone https://github.com/riccardomusmeci/mlx-im
cd mlx-im
pip install .
```
## **Models**
To create a model with weights:
```python
from mlxim.model import create_model
# loading weights from mlx-vision
model = create_model("resnet18")
# loading weights from local file
model = create_model("resnet18", weights="path/to/weights.npz")
```
> [!WARNING]
> As of today (2024-03-05) mlx does not support `nn.Conv2d` with `group` or `dilation` greater than 1 (e.g. `resnext`, `regnet`, `efficientnet`). |