hysts HF staff commited on
Commit
c00e11b
·
1 Parent(s): c63e736

Use PNDM as default scheduler

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -180,8 +180,8 @@ class Model:
180
  return PIL.Image.fromarray(grid)
181
 
182
  def run_simple(self) -> tuple[PIL.Image.Image, PIL.Image.Image]:
183
- self.set_pipeline(self.MODEL_NAMES[0], 'DDIM')
184
- seed = self.rng.randint(0, 1000000)
185
  images = self.generate(seed, num_steps=10, num_images=4)
186
  superresolved = [self.superresolve(image) for image in images]
187
  return self.to_grid(superresolved, 2), self.to_grid(images, 2)
 
180
  return PIL.Image.fromarray(grid)
181
 
182
  def run_simple(self) -> tuple[PIL.Image.Image, PIL.Image.Image]:
183
+ self.set_pipeline(self.MODEL_NAMES[0], 'PNDM')
184
+ seed = self.rng.randint(0, np.iinfo(np.uint32).max + 1)
185
  images = self.generate(seed, num_steps=10, num_images=4)
186
  superresolved = [self.superresolve(image) for image in images]
187
  return self.to_grid(superresolved, 2), self.to_grid(images, 2)