Commit ·
a7e714f
1
Parent(s): b85168a
re-removed yield statements that prevented it running
Browse files- gradio_server.py +1 -7
gradio_server.py
CHANGED
|
@@ -1431,7 +1431,7 @@ def generate_video(
|
|
| 1431 |
end_time = time.time()
|
| 1432 |
abort = True
|
| 1433 |
state["prompt"] = ""
|
| 1434 |
-
|
| 1435 |
else:
|
| 1436 |
sample = samples.cpu()
|
| 1437 |
# video = rearrange(sample.cpu().numpy(), "c t h w -> t h w c")
|
|
@@ -1465,12 +1465,6 @@ def generate_video(
|
|
| 1465 |
|
| 1466 |
print(f"New video saved to Path: "+video_path)
|
| 1467 |
file_list.append(video_path)
|
| 1468 |
-
if video_no < total_video:
|
| 1469 |
-
yield status
|
| 1470 |
-
else:
|
| 1471 |
-
end_time = time.time()
|
| 1472 |
-
state["prompt"] = ""
|
| 1473 |
-
yield f"Total Generation Time: {end_time-start_time:.1f}s"
|
| 1474 |
seed += 1
|
| 1475 |
repeat_no += 1
|
| 1476 |
|
|
|
|
| 1431 |
end_time = time.time()
|
| 1432 |
abort = True
|
| 1433 |
state["prompt"] = ""
|
| 1434 |
+
print(f"Video generation was aborted. Total Generation Time: {end_time-start_time:.1f}s")
|
| 1435 |
else:
|
| 1436 |
sample = samples.cpu()
|
| 1437 |
# video = rearrange(sample.cpu().numpy(), "c t h w -> t h w c")
|
|
|
|
| 1465 |
|
| 1466 |
print(f"New video saved to Path: "+video_path)
|
| 1467 |
file_list.append(video_path)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1468 |
seed += 1
|
| 1469 |
repeat_no += 1
|
| 1470 |
|