akhaliq HF staff commited on
Commit
888af9f
1 Parent(s): f91da70

another fix

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -1171,7 +1171,7 @@ class Model:
1171
  1, 2, 0).to(torch.uint8).numpy()
1172
 
1173
  def run(self, text: str, seed: int,
1174
- only_first_stage: bool,image_prompt: str=None) -> list[np.ndarray]:
1175
  logger.info('==================== run ====================')
1176
  start = time.perf_counter()
1177
 
@@ -1231,7 +1231,7 @@ class AppModel(Model):
1231
 
1232
  def run_with_translation(
1233
  self, text: str, translate: bool, seed: int,
1234
- only_first_stage: bool,image_prompt: str=None) -> tuple[str | None, str | None],
1235
 
1236
  logger.info(f'{text=}, {translate=}, {seed=}, {only_first_stage=},{image_prompt=}')
1237
  if translate:
 
1171
  1, 2, 0).to(torch.uint8).numpy()
1172
 
1173
  def run(self, text: str, seed: int,
1174
+ only_first_stage: bool,image_prompt: None) -> list[np.ndarray]:
1175
  logger.info('==================== run ====================')
1176
  start = time.perf_counter()
1177
 
 
1231
 
1232
  def run_with_translation(
1233
  self, text: str, translate: bool, seed: int,
1234
+ only_first_stage: bool,image_prompt: None) -> tuple[str | None, str | None]:
1235
 
1236
  logger.info(f'{text=}, {translate=}, {seed=}, {only_first_stage=},{image_prompt=}')
1237
  if translate: