fix: dtype

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):

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):

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,

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(