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

more control params #4

Open
jtydhr88 opened this issue Jun 11, 2023 · 0 comments
Open

more control params #4

jtydhr88 opened this issue Jun 11, 2023 · 0 comments

Comments

@jtydhr88
Copy link
Owner

jtydhr88 commented Jun 11, 2023

according to OpenAI Shap-E api, there are more control params can have:

batch_size = 4
guidance_scale = 3.0

image = load_image("example_data/corgi.png")

latents = sample_latents(
    batch_size=batch_size,
    model=model,
    diffusion=diffusion,
    guidance_scale=guidance_scale,
    model_kwargs=dict(images=[image] * batch_size),
    progress=True,
    clip_denoised=True,
    use_fp16=True,
    use_karras=True,
    karras_steps=64,
    sigma_min=1e-3,
    sigma_max=160,
    s_churn=0,
)

Right now, missing the following params, need to add later:
guidance_scale=guidance_scale,
sigma_min=1e-3,
sigma_max=160,
s_churn=0,

@jtydhr88 jtydhr88 mentioned this issue Jun 11, 2023
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