Skip to content

batch_size does not work when it is not 1 #51

Description

@yuxi-liu-wired

There are 5 occurrences of batch_size = 1 in Stable Diffusion Deep Dive.ipynb. The notebook would fail if any of them is set to other values.

The fix is to change

text_input = tokenizer([prompt], ...)
# or
text_input = tokenizer(prompt, ...)

to

text_input = tokenizer([prompt] * batch_size, ...)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions