bharatcoder commited on
Commit
bd32945
1 Parent(s): e4fb45f

Cuda and Seed

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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("mps")
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("mps").manual_seed(12167262721866862)
27
  else:
28
- generator = torch.Generator("mps").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,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=982983949928455, 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"),
 
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"),