Spaces:
Runtime error
Runtime error
bharatcoder
commited on
Commit
•
bd32945
1
Parent(s):
e4fb45f
Cuda and Seed
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def generate_image(prompt, negative_prompt, seed, width, height, guidance_scale,
|
|
15 |
"playgroundai/playground-v2.5-1024px-aesthetic",
|
16 |
torch_dtype=torch.float16,
|
17 |
variant="fp16",
|
18 |
-
).to("
|
19 |
|
20 |
# # Optional: Use DPM++ 2M Karras scheduler for crisper fine details
|
21 |
pipe.scheduler = EDMDPMSolverMultistepScheduler()
|
@@ -23,9 +23,9 @@ def generate_image(prompt, negative_prompt, seed, width, height, guidance_scale,
|
|
23 |
# Check seed
|
24 |
generator = None
|
25 |
if seed == -1:
|
26 |
-
generator = torch.Generator("
|
27 |
else:
|
28 |
-
generator = torch.Generator("
|
29 |
image = pipe(prompt=prompt, num_inference_steps=steps, negative_prompt=negative_prompt, height=height,
|
30 |
generator=generator, width=width, guidance_scale=guidance_scale).images[0]
|
31 |
print('Image generated...')
|
@@ -47,7 +47,7 @@ iface = gr.Interface(
|
|
47 |
inputs=[
|
48 |
gr.Textbox(lines=4, placeholder="Enter your prompt"),
|
49 |
gr.Textbox(lines=4, placeholder="Enter a negative prompt"),
|
50 |
-
gr.Slider(minimum=-1, maximum=
|
51 |
gr.Slider(minimum=720, maximum=1280, value=1024, label="Width"),
|
52 |
gr.Slider(minimum=720, maximum=1280, value=1024, label="Height"),
|
53 |
gr.Slider(minimum=1, maximum=10, value=3, label="Guidance Scale"),
|
|
|
15 |
"playgroundai/playground-v2.5-1024px-aesthetic",
|
16 |
torch_dtype=torch.float16,
|
17 |
variant="fp16",
|
18 |
+
).to("cuda")
|
19 |
|
20 |
# # Optional: Use DPM++ 2M Karras scheduler for crisper fine details
|
21 |
pipe.scheduler = EDMDPMSolverMultistepScheduler()
|
|
|
23 |
# Check seed
|
24 |
generator = None
|
25 |
if seed == -1:
|
26 |
+
generator = torch.Generator("cuda").manual_seed(12167262721866862)
|
27 |
else:
|
28 |
+
generator = torch.Generator("cuda").manual_seed(seed)
|
29 |
image = pipe(prompt=prompt, num_inference_steps=steps, negative_prompt=negative_prompt, height=height,
|
30 |
generator=generator, width=width, guidance_scale=guidance_scale).images[0]
|
31 |
print('Image generated...')
|
|
|
47 |
inputs=[
|
48 |
gr.Textbox(lines=4, placeholder="Enter your prompt"),
|
49 |
gr.Textbox(lines=4, placeholder="Enter a negative prompt"),
|
50 |
+
gr.Slider(minimum=-1, maximum=100000000, value=-1, label="Seed"),
|
51 |
gr.Slider(minimum=720, maximum=1280, value=1024, label="Width"),
|
52 |
gr.Slider(minimum=720, maximum=1280, value=1024, label="Height"),
|
53 |
gr.Slider(minimum=1, maximum=10, value=3, label="Guidance Scale"),
|