search

jina-ai / discoart

🪩 Create Disco Diffusion artworks in one line
Other
3.84k stars 247 forks source link

create ( ... ) always runs 250 steps even when specifying different step amount #168

Open entmike opened 1 year ago

entmike commented 1 year ago

The following command, even though 150 steps is specified, will run 250 steps. DiscoArt version 0.12.0. Not sure if this is because I am passing a custom diffusion_model_config?

da = create(
    text_prompts='a man smiling',
    use_secondary_model=False,
    batch_size=1,
    n_batches=1,
    width_height=[512,512],
    seed=7456,
    steps=150,
    clip_models = [
        "ViT-B-32::openai",
        "ViT-B-16::openai",
        "RN50x64::openai",
        "ViT-L-14-336::openai"
    ],
    diffusion_model="/diffusion-models/finetuned/ffhq/ema_0.9999_105000.pt",
    diffusion_model_config={
        "attention_resolutions": '32, 16, 8',
        "class_cond": False,
        "diffusion_steps": 1000,
        "image_size": 512,
        "learn_sigma": True,
        "noise_schedule": "linear",
        "num_channels": 256,
        "num_head_channels": 64,
        "num_res_blocks": 2,
        "resblock_updown": True,
        "rescale_timesteps": True,
        "timestep_respacing": "250",
        "use_scale_shift_norm": True
    }
)

Screenshot of the 250 bar: image

entmike commented 1 year ago

Figured it out. If I change "timestep_respacing": "250" to a different value, it's playing the role that steps play. Got it. Not sure if this is a bug or not so will leave it open for you to decide.