From 92ac4cb592954799c688f272f67d8f6eb959fe35 Mon Sep 17 00:00:00 2001
From: flozi00 <you@example.com>
Date: Tue, 19 Dec 2023 13:14:02 +0100
Subject: [PATCH] import fix

---
 server/lorax_server/utils/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/server/lorax_server/utils/__init__.py b/server/lorax_server/utils/__init__.py
index 64eec3694..9a5bdaba0 100644
--- a/server/lorax_server/utils/__init__.py
+++ b/server/lorax_server/utils/__init__.py
@@ -20,6 +20,7 @@
     LocalEntryNotFoundError,
     RevisionNotFoundError,
     S3,
+    weight_files,
 )
 from lorax_server.utils.tokens import (
     NextTokenChooser,