(self, tokenizer_path=None)
| 8 | |
| 9 | class SDPrompter(BasePrompter): |
| 10 | def __init__(self, tokenizer_path=None): |
| 11 | if tokenizer_path is None: |
| 12 | base_path = os.path.dirname(os.path.dirname(__file__)) |
| 13 | tokenizer_path = os.path.join(base_path, "tokenizer_configs/stable_diffusion/tokenizer") |
| 14 | super().__init__() |
| 15 | self.tokenizer = CLIPTokenizer.from_pretrained(tokenizer_path) |
| 16 | self.text_encoder: SDTextEncoder = None |
| 17 | self.textual_inversion_dict = {} |
| 18 | self.keyword_dict = {} |
| 19 | |
| 20 | |
| 21 | def fetch_models(self, text_encoder: SDTextEncoder = None): |
nothing calls this directly
no test coverage detected