From 05cbc8488ec17c806079090ccb91203a0c29411b Mon Sep 17 00:00:00 2001 From: Raphael Glon Date: Wed, 29 Nov 2023 17:42:33 +0100 Subject: [PATCH] wip Signed-off-by: Raphael Glon --- docker_images/diffusers/app/pipelines/text_to_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker_images/diffusers/app/pipelines/text_to_image.py b/docker_images/diffusers/app/pipelines/text_to_image.py index bb6530e6..30944ea7 100644 --- 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(