wip

Signed-off-by: Raphael Glon <oOraph@users.noreply.github.com>

docker_images/diffusers/app/pipelines/image_to_image.py

@@ -39,10 +39,12 @@ def __init__(self, model_id: str):
 
         # 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
-        cache_folder_name = file_download.repo_folder_name(
+        cache_root = os.getenv("DIFFUSERS_CACHE", os.getenv("HUGGINGFACE_HUB_CACHE"))
+        folder_name = file_download.repo_folder_name(
             repo_id=model_id, repo_type="model"
         )
-        filename = os.path.join(cache_folder_name, "hub_model_info.json")
+        folder_path = os.path.join(cache_root, folder_name)
+        filename = os.path.join(folder_path, "hub_model_info.json")
         try:
             with open(filename, "r") as f:
                 model_data = json.load(f)

docker_images/diffusers/app/pipelines/text_to_image.py

@@ -30,10 +30,12 @@ def __init__(self, model_id: str):
 
         # 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
-        cache_folder_name = file_download.repo_folder_name(
+        cache_root = os.getenv("DIFFUSERS_CACHE", os.getenv("HUGGINGFACE_HUB_CACHE"))
+        folder_name = file_download.repo_folder_name(
             repo_id=model_id, repo_type="model"
         )
-        filename = os.path.join(cache_folder_name, "hub_model_info.json")
+        folder_path = os.path.join(cache_root, folder_name)
+        filename = os.path.join(folder_path, "hub_model_info.json")
         try:
             with open(filename, "r") as f:
                 model_data = json.load(f)