import fix

predibase · Dec 19, 2023 · 92ac4cb · 92ac4cb
1 parent 0054956
commit 92ac4cb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,