Spaces:
Running
Running
another fix
Browse files
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:
|
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:
|
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:
|