From 2aecbc841e2143e51b04b4a3370b2eb8ae43c451 Mon Sep 17 00:00:00 2001
From: Limitex <76650151+Limitex@users.noreply.github.com>
Date: Sun, 23 Nov 2025 22:57:14 +0900
Subject: [PATCH] fix: dtype

---
 src/usecase/_autoencoder_usecase.py | 2 +-
 src/usecase/_pipeline_usecase.py    | 2 +-
 src/usecase/_sampler_usecase.py     | 2 +-
 src/usecase/_scheduler_usecase.py   | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/usecase/_autoencoder_usecase.py b/src/usecase/_autoencoder_usecase.py
index a0b3cbe..a491491 100644
--- a/src/usecase/_autoencoder_usecase.py
+++ b/src/usecase/_autoencoder_usecase.py
@@ -9,7 +9,7 @@ from ..domain.repositories import AutoencoderRepository
 class AutoencoderUsecase:
     def __init__(self, autoencoder_repo: AutoencoderRepository) -> None:
         self.autoencoder_repo = autoencoder_repo
-        self.dtype = torch.float32
+        self.dtype = torch.float16
 
     def execute(self, vae_path: str) -> Autoencoder:
         if not os.path.exists(vae_path):
diff --git a/src/usecase/_pipeline_usecase.py b/src/usecase/_pipeline_usecase.py
index fda4a1f..c370019 100644
--- a/src/usecase/_pipeline_usecase.py
+++ b/src/usecase/_pipeline_usecase.py
@@ -9,7 +9,7 @@ from ..domain.repositories import PipelineRepository
 class PipelineUsecase:
     def __init__(self, pipeline_repo: PipelineRepository) -> None:
         self.pipeline_repo = pipeline_repo
-        self.dtype = torch.float32
+        self.dtype = torch.float16
 
     def execute(self, checkpoint_path: str) -> tuple[Pipeline, Clip]:
         if not os.path.exists(checkpoint_path):
diff --git a/src/usecase/_sampler_usecase.py b/src/usecase/_sampler_usecase.py
index 780e03c..577b88c 100644
--- a/src/usecase/_sampler_usecase.py
+++ b/src/usecase/_sampler_usecase.py
@@ -19,7 +19,7 @@ from ..domain.repositories import SamplerRepository
 class SamplerUsecase:
     def __init__(self, sampler_repo: SamplerRepository) -> None:
         self.sampler_repo = sampler_repo
-        self.dtype = torch.float32
+        self.dtype = torch.float16
 
     def execute(
         self,
diff --git a/src/usecase/_scheduler_usecase.py b/src/usecase/_scheduler_usecase.py
index 4c9d369..aff50ce 100644
--- a/src/usecase/_scheduler_usecase.py
+++ b/src/usecase/_scheduler_usecase.py
@@ -8,7 +8,7 @@ from ..domain.repositories import SchedulerRepository
 class SchedulerUsecase:
     def __init__(self, scheduler_repo: SchedulerRepository) -> None:
         self.scheduler_repo = scheduler_repo
-        self.dtype = torch.float32
+        self.dtype = torch.float16
 
     def execute(self, pipeline: Pipeline, scheduler_type: Scheduler.Type) -> Scheduler:
         scheduler_obj: SchedulerMixin = self.scheduler_repo.create_scheduler(