mpandzo commited on
Commit
155b00a
·
verified ·
1 Parent(s): dc07cdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -8
app.py CHANGED
@@ -9,13 +9,6 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
9
  torch_dtype=torch.float16,
10
  )
11
 
12
-
13
- seed = None
14
- if seed is None:
15
- seed = int.from_bytes(os.urandom(2), "big")
16
- print(f"Using seed: {seed}")
17
- generator = torch.Generator(device=device).manual_seed(seed)
18
-
19
  def generate_image(prompt):
20
  with torch.device(device):
21
  print(f"Current device: {torch.device(device)}")
@@ -26,7 +19,6 @@ def generate_image(prompt):
26
  num_inference_steps=28,
27
  guidance_scale=7.0,
28
  num_images_per_prompt=1,
29
- generator=generator,
30
  output_type="pil",
31
  return_dict=True,
32
  callback_on_step_end_tensor_inputs=["latents"],
 
9
  torch_dtype=torch.float16,
10
  )
11
 
 
 
 
 
 
 
 
12
  def generate_image(prompt):
13
  with torch.device(device):
14
  print(f"Current device: {torch.device(device)}")
 
19
  num_inference_steps=28,
20
  guidance_scale=7.0,
21
  num_images_per_prompt=1,
 
22
  output_type="pil",
23
  return_dict=True,
24
  callback_on_step_end_tensor_inputs=["latents"],