fix: respect nsfw flag for lora default download

Haidra-Org · Dec 19, 2023 · 2dfba85 · 2dfba85
1 parent 0460a5d
commit 2dfba85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/download_models.py b/download_models.py
@@ -61,7 +61,7 @@ def download_all_models(purge_unused_loras: bool = False) -> None:
         if SharedModelManager.manager.lora is None:
             logger.error("Failed to load LORA model manager")
             exit(1)
-        SharedModelManager.manager.lora.download_default_loras()
+        SharedModelManager.manager.lora.download_default_loras(bridge_data.nsfw)
 
         while SharedModelManager.manager.lora.are_downloads_complete() is False:
             logger.info("Waiting for LORA downloads to complete...")