callum-canavan commited on
Commit
cf5c20f
·
1 Parent(s): dba8464

Add options

Browse files
Files changed (1) hide show
  1. app.py +15 -4
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,