From 2bf11c463318f83f785a67bf7b7d573cba7ac57c Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Fri, 3 May 2024 17:13:05 -0700
Subject: [PATCH] Use stable url for MNIST (#749)

* use stable url

* remove deprecated flag
---
 llms/mlx_lm/utils.py | 1 -
 mnist/mnist.py       | 4 +++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/llms/mlx_lm/utils.py b/llms/mlx_lm/utils.py
index 3acc5a62..5a190b7d 100644
--- a/llms/mlx_lm/utils.py
+++ b/llms/mlx_lm/utils.py
@@ -80,7 +80,6 @@ def get_model_path(path_or_hf_repo: str, revision: Optional[str] = None) -> Path
                     "*.tiktoken",
                     "*.txt",
                 ],
-                resume_download=True,
             )
         )
     return model_path
diff --git a/mnist/mnist.py b/mnist/mnist.py
index 1a74fdd7..c5f920e6 100644
--- a/mnist/mnist.py
+++ b/mnist/mnist.py
@@ -9,7 +9,9 @@ import numpy as np
 
 
 def mnist(
-    save_dir="/tmp", base_url="http://yann.lecun.com/exdb/mnist/", filename="mnist.pkl"
+    save_dir="/tmp",
+    base_url="https://raw.githubusercontent.com/fgnt/mnist/master/",
+    filename="mnist.pkl",
 ):
     """
     Load the MNIST dataset in 4 tensors: train images, train labels,