Spaces:
Running
on
Zero
Running
on
Zero
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:
|