diff --git a/tabby/tools/download_models.py b/tabby/tools/download_models.py
index 89127e1..26a065b 100644
--- a/tabby/tools/download_models.py
+++ b/tabby/tools/download_models.py
@@ -40,7 +40,7 @@ if __name__ == "__main__":
     try:
         preload(local_files_only=args.prefer_local_files)
     except Exception as e:
-        if "offline" in str(e):
+        if "offline" in str(e) or "local_files_only" in str(e):
             preload(local_files_only=False)
         else:
             raise e