Tophness2022 commited on
Commit
a7e714f
·
1 Parent(s): b85168a

re-removed yield statements that prevented it running

Browse files
Files changed (1) hide show
  1. 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
- yield 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,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