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,