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)
|