Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
entmike opened this issue Aug 16, 2022 · 1 comment

Comments

@entmike
Copy link

entmike commented Aug 16, 2022

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
Copy link
Author

entmike commented Aug 17, 2022

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant