Spaces:
Runtime error
Runtime error
Use PNDM as default scheduler
Browse files
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], '
|
184 |
-
seed = self.rng.randint(0,
|
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)
|