(self, model_name, is_vl=False, device=0, **kwargs)
| 164 | class PromptExpander: |
| 165 | |
| 166 | def __init__(self, model_name, is_vl=False, device=0, **kwargs): |
| 167 | self.model_name = model_name |
| 168 | self.is_vl = is_vl |
| 169 | self.device = device |
| 170 | |
| 171 | def extend_with_img(self, |
| 172 | prompt, |