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