hysts HF staff commited on
Commit
53ddf51
1 Parent(s): fce8c40

Fix type annotation

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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 | None]:
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))