mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-24 09:21:18 +08:00
docs: added missing imports (#375)
* add: missing import * add: missing import
This commit is contained in:
parent
b1dec281b3
commit
0f19237fb8
@ -23,6 +23,7 @@ python convert.py \
|
||||
To use the `Bert` model in your own code, you can load it with:
|
||||
|
||||
```python
|
||||
import mlx.core as mx
|
||||
from model import Bert, load_model
|
||||
|
||||
model, tokenizer = load_model(
|
||||
|
@ -31,6 +31,7 @@ way to get started is by using the `StableDiffusion` class from the `stable_diff
|
||||
module.
|
||||
|
||||
```python
|
||||
import mlx.core as mx
|
||||
from stable_diffusion import StableDiffusion
|
||||
|
||||
# This will download all the weights from HF hub and load the models in
|
||||
|
Loading…
Reference in New Issue
Block a user