From 0f19237fb8697e2551197ccb5c990a429f6ba048 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Yi=C4=9Fit=20=C3=96=2E=20=C3=9Cnver?=
 <69584310+younver@users.noreply.github.com>
Date: Thu, 25 Jan 2024 21:44:53 +0300
Subject: [PATCH] docs: added missing imports (#375)

* add: missing import

* add: missing import
---
 bert/README.md             | 1 +
 stable_diffusion/README.md | 1 +
 2 files changed, 2 insertions(+)

diff --git a/bert/README.md b/bert/README.md
index 42e5e957..e03df7d0 100644
--- a/bert/README.md
+++ b/bert/README.md
@@ -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(
diff --git a/stable_diffusion/README.md b/stable_diffusion/README.md
index 385ce6d9..711ec10c 100644
--- a/stable_diffusion/README.md
+++ b/stable_diffusion/README.md
@@ -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