From 65f4968e5f4869332cbdc3e3b6bf0b3cd9e956e7 Mon Sep 17 00:00:00 2001
From: Anthony DePasquale <anthony@depasquale.org>
Date: Tue, 28 May 2024 23:05:37 +0200
Subject: [PATCH] Fix download progress (#78)

---
 Libraries/LLM/Load.swift | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Libraries/LLM/Load.swift b/Libraries/LLM/Load.swift
index e7716ac..4d9849a 100644
--- a/Libraries/LLM/Load.swift
+++ b/Libraries/LLM/Load.swift
@@ -24,9 +24,9 @@ public func load(
 
         switch configuration.id {
         case .id(let id):
-            // download the model weights and config
+            // download the model weights
             let repo = Hub.Repo(id: id)
-            let modelFiles = ["config.json", "*.safetensors"]
+            let modelFiles = ["*.safetensors"]
             modelDirectory = try await hub.snapshot(
                 from: repo, matching: modelFiles, progressHandler: progressHandler)