Spaces:
Runtime error
Runtime error
Fix type annotation
Browse files
model.py
CHANGED
|
@@ -144,7 +144,7 @@ class Model:
|
|
| 144 |
|
| 145 |
def run(self, model_name: str, scheduler_type: str, num_steps: int,
|
| 146 |
randomize_seed: bool,
|
| 147 |
-
seed: int) -> tuple[PIL.Image.Image, int, str
|
| 148 |
self.set_pipeline(model_name, scheduler_type)
|
| 149 |
if scheduler_type == 'PNDM':
|
| 150 |
num_steps = max(4, min(num_steps, 100))
|
|
|
|
| 144 |
|
| 145 |
def run(self, model_name: str, scheduler_type: str, num_steps: int,
|
| 146 |
randomize_seed: bool,
|
| 147 |
+
seed: int) -> tuple[PIL.Image.Image, int, str]:
|
| 148 |
self.set_pipeline(model_name, scheduler_type)
|
| 149 |
if scheduler_type == 'PNDM':
|
| 150 |
num_steps = max(4, min(num_steps, 100))
|