Update app.py
Browse files
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"],
|