wip

Signed-off-by: Raphael Glon <oOraph@users.noreply.github.com>
huggingface · Nov 29, 2023 · 05cbc84 · 05cbc84
1 parent 29270b8
commit 05cbc84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker_images/diffusers/app/pipelines/text_to_image.py b/docker_images/diffusers/app/pipelines/text_to_image.py
@@ -29,7 +29,7 @@ def __init__(self, model_id: str):
         self.use_auth_token = os.getenv("HF_API_TOKEN")
         # This should allow us to make the image work with private models when no token is provided, if the said model
         # is already in local cache
-        self.offline_preferred = os.getenv("OFFLINE_PREFERRED")
+        self.offline_preferred = validation.str_to_bool(os.getenv("OFFLINE_PREFERRED"))
         fetched = False
         if self.offline_preferred:
             cache_root = os.getenv(