HelloSun commited on
Commit
5473b5a
·
verified ·
1 Parent(s): 7006304

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -21,19 +21,19 @@ MAX_IMAGE_SIZE = 256
21
 
22
  def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
23
 
24
- if randomize_seed:
25
- seed = random.randint(0, MAX_SEED)
26
 
27
- generator = torch.Generator().manual_seed(seed)
28
 
29
  image = pipe(
30
  prompt = prompt,
31
  negative_prompt = negative_prompt,
32
- guidance_scale = guidance_scale,
33
- num_inference_steps = num_inference_steps,
34
  width = width,
35
  height = height,
36
- generator = generator
37
  ).images[0]
38
 
39
  return image
 
21
 
22
  def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
23
 
24
+ #if randomize_seed:
25
+ # seed = random.randint(0, MAX_SEED)
26
 
27
+ #generator = torch.Generator().manual_seed(seed)
28
 
29
  image = pipe(
30
  prompt = prompt,
31
  negative_prompt = negative_prompt,
32
+ #guidance_scale = guidance_scale,
33
+ #num_inference_steps = num_inference_steps,
34
  width = width,
35
  height = height,
36
+ #generator = generator
37
  ).images[0]
38
 
39
  return image