fix: dtype

2025-11-23 22:57:53 +09:00
parent e17bdfc621
commit 2aecbc841e
4 changed files with 4 additions and 4 deletions
@@ -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,