From 73e4c8dbdab552234d76a1f5a54f05842f3690df Mon Sep 17 00:00:00 2001
From: Limitex <76650151+Limitex@users.noreply.github.com>
Date: Sun, 9 Nov 2025 06:57:09 +0900
Subject: [PATCH] fix: cache path for diffusers repository cache

---
 src/infra/diffusers/_autoencoder_repository.py | 4 ++--
 src/infra/diffusers/_pipeline_repository.py    | 4 ++--
 src/infra/diffusers/_scheduler_repository.py   | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/infra/diffusers/_autoencoder_repository.py b/src/infra/diffusers/_autoencoder_repository.py
index 2337a84..804474d 100644
--- a/src/infra/diffusers/_autoencoder_repository.py
+++ b/src/infra/diffusers/_autoencoder_repository.py
@@ -2,7 +2,6 @@ import io
 import os
 from typing import Any
 
-import folder_paths  # pyright: ignore[reportMissingImports]
 import requests
 import torch
 from comfy.model_management import get_torch_device  # pyright: ignore[reportMissingImports]
@@ -18,12 +17,13 @@ from omegaconf import OmegaConf
 from safetensors import safe_open
 
 from ...domain.repositories import AutoencoderRepository
+from ...utils import get_cache_path
 from ._cache import is_vae_cached
 
 
 class DiffusersAutoencoderRepository(AutoencoderRepository):
     def __init__(self) -> None:
-        self.cache_dir = folder_paths.get_temp_directory()
+        self.cache_dir = get_cache_path()
         self.device = get_torch_device()
 
     # Reference from : https://github.com/huggingface/diffusers/blob/main/scripts/convert_vae_pt_to_diffusers.py
diff --git a/src/infra/diffusers/_pipeline_repository.py b/src/infra/diffusers/_pipeline_repository.py
index 8322176..6485e8c 100644
--- a/src/infra/diffusers/_pipeline_repository.py
+++ b/src/infra/diffusers/_pipeline_repository.py
@@ -1,17 +1,17 @@
 import os
 
-import folder_paths  # pyright: ignore[reportMissingImports]
 import torch
 from comfy.model_management import get_torch_device  # pyright: ignore[reportMissingImports]
 from diffusers import StableDiffusionPipeline
 
 from ...domain.repositories import PipelineRepository
+from ...utils import get_cache_path
 from ._cache import is_pipeline_cached
 
 
 class DiffusersPipelineRepository(PipelineRepository):
     def __init__(self) -> None:
-        self.cache_dir = folder_paths.get_temp_directory()
+        self.cache_dir = get_cache_path()
         self.device = get_torch_device()
 
     def convert_and_save_from_single_file(self, checkpoint_path: str, dtype: torch.dtype) -> str:
diff --git a/src/infra/diffusers/_scheduler_repository.py b/src/infra/diffusers/_scheduler_repository.py
index 4ccc834..b82e530 100644
--- a/src/infra/diffusers/_scheduler_repository.py
+++ b/src/infra/diffusers/_scheduler_repository.py
@@ -1,14 +1,14 @@
-import folder_paths  # pyright: ignore[reportMissingImports]
 import torch
 from diffusers.schedulers.scheduling_utils import SchedulerMixin
 
 from ...domain.model import Scheduler
 from ...domain.repositories import SchedulerRepository
+from ...utils import get_cache_path
 
 
 class DiffusersSchedulerRepository(SchedulerRepository):
     def __init__(self) -> None:
-        self.cache_dir = folder_paths.get_temp_directory()
+        self.cache_dir = get_cache_path()
 
     def create_scheduler(
         self, model_path: str, dtype: torch.dtype, scheduler_type: Scheduler.Type