From f452c56ea69d57d69ef99f563242ced8f73d6176 Mon Sep 17 00:00:00 2001 From: corkborg <62742115+corkborg@users.noreply.github.com> Date: Tue, 23 Apr 2024 01:20:42 +0900 Subject: [PATCH] Remove useless code. --- tagger/interrogator.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tagger/interrogator.py b/tagger/interrogator.py index e85112f..fa88b9d 100644 --- a/tagger/interrogator.py +++ b/tagger/interrogator.py @@ -114,7 +114,6 @@ class WaifuDiffusionInterrogator(Interrogator): self.model_path = model_path self.tags_path = tags_path self.kwargs = kwargs - self.use_cpu = True def download(self) -> Tuple[os.PathLike, os.PathLike]: print(f"Loading {self.name} model file from {self.kwargs['repo_id']}") @@ -199,7 +198,6 @@ class MLDanbooruInterrogator(Interrogator): self.repo_id = repo_id self.tags = None self.model = None - self.use_cpu = False def download(self) -> Tuple[str, str]: print(f"Loading {self.name} model file from {self.repo_id}")