hysts HF Staff commited on
Commit
6a99664
·
1 Parent(s): a8442ab
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -20,8 +20,8 @@ MAX_IMAGE_SIZE = 2048
20
  @spaces.GPU(duration=75)
21
  def infer(
22
  prompt: str,
23
- seed: int = 42,
24
- randomize_seed: bool = False,
25
  width: int = 1024,
26
  height: int = 1024,
27
  guidance_scale: float = 3.5,
@@ -32,13 +32,13 @@ def infer(
32
 
33
  Args:
34
  prompt: The prompt to generate an image from.
35
- seed: The seed to use for the image generation. Defaults to 42.
36
- randomize_seed: Whether to randomize the seed. Defaults to False.
37
  width: The width of the image. Defaults to 1024.
38
  height: The height of the image. Defaults to 1024.
39
  guidance_scale: The guidance scale to use for the image generation. Defaults to 3.5.
40
  num_inference_steps: The number of inference steps to use for the image generation. Defaults to 28.
41
- progress: The progress bar to use for the image generation. Defaults to a progress bar that tracks the tqdm progress.
42
 
43
  Returns:
44
  A tuple containing the generated image and the seed.
 
20
  @spaces.GPU(duration=75)
21
  def infer(
22
  prompt: str,
23
+ seed: int,
24
+ randomize_seed: bool,
25
  width: int = 1024,
26
  height: int = 1024,
27
  guidance_scale: float = 3.5,
 
32
 
33
  Args:
34
  prompt: The prompt to generate an image from.
35
+ seed: The seed to use for the image generation.
36
+ randomize_seed: Whether to randomize the seed.
37
  width: The width of the image. Defaults to 1024.
38
  height: The height of the image. Defaults to 1024.
39
  guidance_scale: The guidance scale to use for the image generation. Defaults to 3.5.
40
  num_inference_steps: The number of inference steps to use for the image generation. Defaults to 28.
41
+ progress: Internal parameter used to display progress in the UI. This should not be set manually by the user.
42
 
43
  Returns:
44
  A tuple containing the generated image and the seed.