K00B404 commited on
Commit
6f22587
·
verified ·
1 Parent(s): d0c42ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -33,7 +33,7 @@ pipe = FluxPipeline.from_pretrained(
33
  prompt = "A close-up image of a green alien with fluorescent skin in the middle of a dark purple forest"
34
 
35
  guidance_scale = 3.5 # Keep guidance_scale at 3.5
36
- n_steps = 28
37
  seed = 11
38
 
39
  with torch.inference_mode():
@@ -42,8 +42,8 @@ with torch.inference_mode():
42
  generator=torch.Generator(device="cpu").manual_seed(seed),
43
  num_inference_steps=n_steps,
44
  guidance_scale=guidance_scale,
45
- height=1024,
46
- width=1024,
47
  ).images[0]
48
  image.save("output.png")
49
 
 
33
  prompt = "A close-up image of a green alien with fluorescent skin in the middle of a dark purple forest"
34
 
35
  guidance_scale = 3.5 # Keep guidance_scale at 3.5
36
+ n_steps = 8
37
  seed = 11
38
 
39
  with torch.inference_mode():
 
42
  generator=torch.Generator(device="cpu").manual_seed(seed),
43
  num_inference_steps=n_steps,
44
  guidance_scale=guidance_scale,
45
+ height=512,
46
+ width=512,
47
  ).images[0]
48
  image.save("output.png")
49