docs: added missing imports (#375)

* add: missing import

* add: missing import
This commit is contained in:
Yiğit Ö. Ünver 2024-01-25 21:44:53 +03:00 committed by GitHub
parent b1dec281b3
commit 0f19237fb8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 0 deletions

View File

@ -23,6 +23,7 @@ python convert.py \
To use the `Bert` model in your own code, you can load it with: To use the `Bert` model in your own code, you can load it with:
```python ```python
import mlx.core as mx
from model import Bert, load_model from model import Bert, load_model
model, tokenizer = load_model( model, tokenizer = load_model(

View File

@ -31,6 +31,7 @@ way to get started is by using the `StableDiffusion` class from the `stable_diff
module. module.
```python ```python
import mlx.core as mx
from stable_diffusion import StableDiffusion from stable_diffusion import StableDiffusion
# This will download all the weights from HF hub and load the models in # This will download all the weights from HF hub and load the models in