(
self,
tokenizer_1_path=None,
tokenizer_2_path=None,
tokenizer_3_path=None
)
| 7 | |
| 8 | class SD3Prompter(BasePrompter): |
| 9 | def __init__( |
| 10 | self, |
| 11 | tokenizer_1_path=None, |
| 12 | tokenizer_2_path=None, |
| 13 | tokenizer_3_path=None |
| 14 | ): |
| 15 | if tokenizer_1_path is None: |
| 16 | base_path = os.path.dirname(os.path.dirname(__file__)) |
| 17 | tokenizer_1_path = os.path.join(base_path, "tokenizer_configs/stable_diffusion_3/tokenizer_1") |
| 18 | if tokenizer_2_path is None: |
| 19 | base_path = os.path.dirname(os.path.dirname(__file__)) |
| 20 | tokenizer_2_path = os.path.join(base_path, "tokenizer_configs/stable_diffusion_3/tokenizer_2") |
| 21 | if tokenizer_3_path is None: |
| 22 | base_path = os.path.dirname(os.path.dirname(__file__)) |
| 23 | tokenizer_3_path = os.path.join(base_path, "tokenizer_configs/stable_diffusion_3/tokenizer_3") |
| 24 | super().__init__() |
| 25 | self.tokenizer_1 = CLIPTokenizer.from_pretrained(tokenizer_1_path) |
| 26 | self.tokenizer_2 = CLIPTokenizer.from_pretrained(tokenizer_2_path) |
| 27 | self.tokenizer_3 = T5TokenizerFast.from_pretrained(tokenizer_3_path) |
| 28 | self.text_encoder_1: SD3TextEncoder1 = None |
| 29 | self.text_encoder_2: SD3TextEncoder2 = None |
| 30 | self.text_encoder_3: SD3TextEncoder3 = None |
| 31 | |
| 32 | |
| 33 | def fetch_models(self, text_encoder_1: SD3TextEncoder1 = None, text_encoder_2: SD3TextEncoder2 = None, text_encoder_3: SD3TextEncoder3 = None): |
nothing calls this directly
no test coverage detected