Commit
·
cf5c20f
1
Parent(s):
dba8464
Add options
Browse files
app.py
CHANGED
@@ -31,7 +31,10 @@ def generate_content(
|
|
31 |
prompt_for_transformed,
|
32 |
transformation,
|
33 |
num_inference_steps,
|
34 |
-
seed
|
|
|
|
|
|
|
35 |
):
|
36 |
prompts = [f'{style} {p}'.strip() for p in [prompt_for_original, prompt_for_transformed]]
|
37 |
prompt_embeds = [stage_1.encode_prompt(p) for p in prompts]
|
@@ -50,7 +53,9 @@ def generate_content(
|
|
50 |
negative_prompt_embeds,
|
51 |
views,
|
52 |
num_inference_steps=num_inference_steps,
|
53 |
-
generator=generator
|
|
|
|
|
54 |
|
55 |
print("Sample stage 2")
|
56 |
image = sample_stage_2(stage_2,
|
@@ -59,7 +64,10 @@ def generate_content(
|
|
59 |
negative_prompt_embeds,
|
60 |
views,
|
61 |
num_inference_steps=num_inference_steps,
|
62 |
-
generator=generator
|
|
|
|
|
|
|
63 |
save_illusion(image, views, Path(""))
|
64 |
|
65 |
output_name = f"illusion.mp4"
|
@@ -87,7 +95,10 @@ gradio_app = gr.Interface(
|
|
87 |
gr.Textbox(label="Prompt for transformed view", placeholder="an old man"),
|
88 |
gr.Dropdown(label="View transformation", choices=choices, value=choices[0]),
|
89 |
gr.Number(label="Number of diffusion steps", value=75, step=1, minimum=1, maximum=300),
|
90 |
-
gr.Number(label="Random seed", value=0, step=1, minimum=0, maximum=100000)
|
|
|
|
|
|
|
91 |
],
|
92 |
outputs=[gr.Video(label="Illusion"), gr.Image(label="Before and After")],
|
93 |
description=description,
|
|
|
31 |
prompt_for_transformed,
|
32 |
transformation,
|
33 |
num_inference_steps,
|
34 |
+
seed,
|
35 |
+
guidance_scale,
|
36 |
+
noise_level,
|
37 |
+
reduction,
|
38 |
):
|
39 |
prompts = [f'{style} {p}'.strip() for p in [prompt_for_original, prompt_for_transformed]]
|
40 |
prompt_embeds = [stage_1.encode_prompt(p) for p in prompts]
|
|
|
53 |
negative_prompt_embeds,
|
54 |
views,
|
55 |
num_inference_steps=num_inference_steps,
|
56 |
+
generator=generator,
|
57 |
+
guidance_scale=guidance_scale,
|
58 |
+
reduction=reduction)
|
59 |
|
60 |
print("Sample stage 2")
|
61 |
image = sample_stage_2(stage_2,
|
|
|
64 |
negative_prompt_embeds,
|
65 |
views,
|
66 |
num_inference_steps=num_inference_steps,
|
67 |
+
generator=generator,
|
68 |
+
guidance_scale=guidance_scale,
|
69 |
+
noise_level=noise_level,
|
70 |
+
reduction=reduction)
|
71 |
save_illusion(image, views, Path(""))
|
72 |
|
73 |
output_name = f"illusion.mp4"
|
|
|
95 |
gr.Textbox(label="Prompt for transformed view", placeholder="an old man"),
|
96 |
gr.Dropdown(label="View transformation", choices=choices, value=choices[0]),
|
97 |
gr.Number(label="Number of diffusion steps", value=75, step=1, minimum=1, maximum=300),
|
98 |
+
gr.Number(label="Random seed", value=0, step=1, minimum=0, maximum=100000),
|
99 |
+
gr.Number(label="Guidance scale", value=7.0, step=0.1, minimum=0.1, maximum=10.0, info="How much diffusion is guided by the prompt"),
|
100 |
+
gr.Number(label="Noise level", value=50, step=1, minimum=1, maximum=100, info="Noise level for stage 2 of diffusion sampling")
|
101 |
+
gr.Dropdown(label="Reduction", choices=["mean", "alternate"], value="mean", info="How to reduce predicted noise and variances")
|
102 |
],
|
103 |
outputs=[gr.Video(label="Illusion"), gr.Image(label="Before and After")],
|
104 |
description=description,
|