[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
1cfb516d1b
commit
7c9521810e
27 changed files with 298 additions and 196 deletions
|
|
@ -135,7 +135,12 @@ def test_lora_attached_bypasses_the_cache(cache_env):
|
|||
class _ListEncodePipe(_EncodePipe):
|
||||
"""Returns per-prompt embedding LISTS like Z-Image's ``encode_prompt``."""
|
||||
|
||||
def encode_prompt(self, prompt, device = None, do_classifier_free_guidance = True):
|
||||
def encode_prompt(
|
||||
self,
|
||||
prompt,
|
||||
device = None,
|
||||
do_classifier_free_guidance = True,
|
||||
):
|
||||
self.calls += 1
|
||||
prompts = prompt if isinstance(prompt, list) else [prompt]
|
||||
embeds = [torch.full((1, 4), float(sum(map(ord, p)))) for p in prompts]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue