johnrachwanpruna commited on
Commit
5d9cf5d
·
verified ·
1 Parent(s): 22af02b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ MAX_SEED = np.iinfo(np.int32).max
19
  MAX_IMAGE_SIZE = 2048
20
 
21
  @spaces.GPU()
22
- def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
23
  if randomize_seed:
24
  seed = random.randint(0, MAX_SEED)
25
  generator = torch.Generator().manual_seed(seed)
 
19
  MAX_IMAGE_SIZE = 2048
20
 
21
  @spaces.GPU()
22
+ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
23
  if randomize_seed:
24
  seed = random.randint(0, MAX_SEED)
25
  generator = torch.Generator().manual_seed(seed)