sdart commited on
Commit
1e0e2d1
1 Parent(s): 1ecf68a

update steps

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -19,7 +19,7 @@ class Model:
19
  negative_prompt: str,
20
  guidance_scale:int = 7,
21
  num_images:int = 1,
22
- num_steps:int = 2,
23
  ):
24
  seed = np.random.randint(0, np.iinfo(np.int64).max)
25
  generator = torch.Generator().manual_seed(seed)
 
19
  negative_prompt: str,
20
  guidance_scale:int = 7,
21
  num_images:int = 1,
22
+ num_steps:int = 20,
23
  ):
24
  seed = np.random.randint(0, np.iinfo(np.int64).max)
25
  generator = torch.Generator().manual_seed(seed)