Skip to content

Inference Parameters #5

Description

@Tomato1101
with torch.inference_mode():
    seed = 3407+j
    prompt = data['caption'][0]
    idx = data['idx'][0]
    image = pipe(
        prompt,
        height=1024,
        width=1024,
        guidance_scale=3.5,
        num_inference_steps=30,
        max_sequence_length=512,
        generator=torch.Generator(device="cuda").manual_seed(seed)
    ).images[0]

针对leaderboard上开源模型的分数有一个问题请教一下,在第一步推理生成图像的时候,开源模型全部都是基于flux_zh_multi_node_inference.py中上述推理参数生成的,还是每个开源模型各自采用了官方的原始推理参数?

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions